malavikapradeep2001 commited on
Commit
50ad802
·
unverified ·
1 Parent(s): 84fb3c8

Update backend/app.py

Browse files
Files changed (1) hide show
  1. backend/app.py +5 -0
backend/app.py CHANGED
@@ -18,6 +18,11 @@ from model_histo import BreastCancerClassifier # TensorFlow model
18
  from huggingface_hub import login
19
  import os
20
 
 
 
 
 
 
21
  hf_token = os.getenv("HF_TOKEN")
22
  if hf_token:
23
  login(token=hf_token)
 
18
  from huggingface_hub import login
19
  import os
20
 
21
+ os.environ["TORCH_HOME"] = "/tmp/torch"
22
+ os.environ["HF_HOME"] = "/tmp/huggingface"
23
+ os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
24
+ os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
25
+
26
  hf_token = os.getenv("HF_TOKEN")
27
  if hf_token:
28
  login(token=hf_token)