Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,8 @@ async def load_model():
|
|
27 |
|
28 |
# Load model (will cache in /home/user/.cache/huggingface)
|
29 |
app.state.model = QwenClassifier.from_pretrained(
|
30 |
-
'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446'
|
|
|
31 |
)
|
32 |
print("Model loaded successfully!")
|
33 |
|
|
|
27 |
|
28 |
# Load model (will cache in /home/user/.cache/huggingface)
|
29 |
app.state.model = QwenClassifier.from_pretrained(
|
30 |
+
'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446',
|
31 |
+
cache_dir=os.environ['HF_HOME'] # Explicit cache location
|
32 |
)
|
33 |
print("Model loaded successfully!")
|
34 |
|