Update app.py
Browse files
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 |
-
|
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")
|