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