Spaces:

KeivanR
/

qwen-classifier-demo

Sleeping

KeivanR commited on Mar 29

Commit

d3eff8a

verified ·

1 Parent(s): d394f04

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,6 @@
 from fastapi import FastAPI
 from qwen_classifier.predict import predict_single  # Your existing function
 import torch

+os.environ['HF_HOME'] = '/home/user/.cache/huggingface'
+os.environ['TRANSFORMERS_CACHE'] = os.environ['HF_HOME']
 from fastapi import FastAPI
 from qwen_classifier.predict import predict_single  # Your existing function
 import torch