Spaces:

KeivanR
/

qwen-classifier-demo

Sleeping

KeivanR commited on Mar 29

Commit

6e9d355

verified ·

1 Parent(s): 318285f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import os
-os.environ['HF_HOME'] = '/home/user/.cache/huggingface'
 os.environ['TRANSFORMERS_CACHE'] = os.environ['HF_HOME']
 from fastapi import FastAPI

 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