malavikapradeep2001 commited on
Commit
f20a61d
·
unverified ·
1 Parent(s): 523131c

Update backend/app.py

Browse files
Files changed (1) hide show
  1. backend/app.py +14 -13
backend/app.py CHANGED
@@ -1,4 +1,18 @@
1
  import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  from fastapi import FastAPI, File, UploadFile, Form
3
  from fastapi.middleware.cors import CORSMiddleware
4
  from fastapi.responses import JSONResponse
@@ -16,19 +30,6 @@ from model import MWT as create_model
16
  from augmentations import Augmentations
17
  from model_histo import BreastCancerClassifier # TensorFlow model
18
 
19
- os.environ["HF_HOME"] = "/tmp/huggingface"
20
- os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
21
- os.environ["TORCH_HOME"] = "/tmp/torch"
22
- os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
23
- os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
24
-
25
- from huggingface_hub import login
26
-
27
-
28
-
29
- hf_token = os.getenv("HF_TOKEN")
30
- if hf_token:
31
- login(token=hf_token)
32
 
33
 
34
  # =====================================================
 
1
  import os
2
+ os.environ["HF_HOME"] = "/tmp/huggingface"
3
+ os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
4
+ os.environ["TORCH_HOME"] = "/tmp/torch"
5
+ os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
6
+ os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
7
+
8
+ from huggingface_hub import login
9
+
10
+
11
+
12
+ hf_token = os.getenv("HF_TOKEN")
13
+ if hf_token:
14
+ login(token=hf_token)
15
+
16
  from fastapi import FastAPI, File, UploadFile, Form
17
  from fastapi.middleware.cors import CORSMiddleware
18
  from fastapi.responses import JSONResponse
 
30
  from augmentations import Augmentations
31
  from model_histo import BreastCancerClassifier # TensorFlow model
32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
 
34
 
35
  # =====================================================