abhicodes commited on
Commit
d9695c5
1 Parent(s): 275319c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ API_KEY = os.getenv("API_KEY")
18
  # BRAIN_TUMOR_API_URL = "https://api-inference.huggingface.co/models/Devarshi/Brain_Tumor_Classification"
19
  BREAST_CANCER_API_URL = "https://api-inference.huggingface.co/models/MUmairAB/Breast_Cancer_Detector"
20
  # ALZHEIMER_API_URL = "https://api-inference.huggingface.co/models/AhmadHakami/alzheimer-image-classification-google-vit-base-patch16"
21
- # headers = {"Authorization": "Bearer "+API_KEY+"", 'Content-Type': 'application/json'}
22
  alzheimer_classifier = pipeline("image-classification", model="AhmadHakami/alzheimer-image-classification-google-vit-base-patch16")
23
  # breast_cancer_classifier = pipeline("image-classification", model="MUmairAB/Breast_Cancer_Detector")
24
  brain_tumor_classifier = pipeline("image-classification", model="Devarshi/Brain_Tumor_Classification")
 
18
  # BRAIN_TUMOR_API_URL = "https://api-inference.huggingface.co/models/Devarshi/Brain_Tumor_Classification"
19
  BREAST_CANCER_API_URL = "https://api-inference.huggingface.co/models/MUmairAB/Breast_Cancer_Detector"
20
  # ALZHEIMER_API_URL = "https://api-inference.huggingface.co/models/AhmadHakami/alzheimer-image-classification-google-vit-base-patch16"
21
+ headers = {"Authorization": "Bearer "+API_KEY+"", 'Content-Type': 'application/json'}
22
  alzheimer_classifier = pipeline("image-classification", model="AhmadHakami/alzheimer-image-classification-google-vit-base-patch16")
23
  # breast_cancer_classifier = pipeline("image-classification", model="MUmairAB/Breast_Cancer_Detector")
24
  brain_tumor_classifier = pipeline("image-classification", model="Devarshi/Brain_Tumor_Classification")