abhicodes commited on
Commit
199de28
·
verified ·
1 Parent(s): 6233aa1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="AhmadHakami/alzheimer-image-classification-google-vit-base-patch16")
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