KeivanR commited on
Commit
8bb21ff
·
verified ·
1 Parent(s): 6e9d355

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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