dewifaj commited on
Commit
69f1ee6
1 Parent(s): 52628b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
  def alz_mri_classification(image):
4
- classifier = pipeline("image-classification", model="dewifaj/resnet18_alzheimer_classifier")
5
  result = classifier(image)
6
  # extract the highest score
7
  prediction = result[0]
 
1
  from transformers import pipeline
2
  import gradio as gr
3
  def alz_mri_classification(image):
4
+ classifier = pipeline("image-classification", model="dewifaj/alzheimer_mri_classification")
5
  result = classifier(image)
6
  # extract the highest score
7
  prediction = result[0]