Update backend/app.py

backend/app.py

@@ -15,14 +15,16 @@ from model import MWT as create_model
 from augmentations import Augmentations
 from model_histo import BreastCancerClassifier  # TensorFlow model
 
-from huggingface_hub import login
-import os
-
-os.environ["TORCH_HOME"] = "/tmp/torch"
 os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
+os.environ["TORCH_HOME"] = "/tmp/torch"
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
 os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
 
+from huggingface_hub import login
+import os
+
+
 hf_token = os.getenv("HF_TOKEN")
 if hf_token:
     login(token=hf_token)