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")
|