Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 
 os.environ['HF_HOME'] = '/tmp/.cache/huggingface'  # Use /tmp in Spaces
 os.makedirs(os.environ['HF_HOME'], exist_ok=True)  # Ensure directory exists
-os.environ['TRANSFORMERS_CACHE'] = os.environ['HF_HOME'] 
 
 from fastapi import FastAPI
 from qwen_classifier.predict import predict_single  # Your existing function
@@ -28,7 +27,6 @@ async def load_model():
     # Load model (will cache in /home/user/.cache/huggingface)
     app.state.model = QwenClassifier.from_pretrained(
         'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446',
-            cache_dir=os.environ['HF_HOME']  # Explicit cache location
     )
     print("Model loaded successfully!")