abhicodes commited on
Commit
9f394ff
1 Parent(s): 0fa87b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -31,10 +31,9 @@ alzheimer_classifier = pipeline("image-classification", model="dewifaj/alzheimer
31
  # Create a function to Detect/Classify Alzheimer
32
  def classify_alzheimer(image):
33
  result = alzheimer_classifier(image)
34
- print(result)
35
  prediction = {}
36
  for ele in result:
37
- score, label = ele.values()
38
  prediction[label] = score
39
  return prediction
40
 
 
31
  # Create a function to Detect/Classify Alzheimer
32
  def classify_alzheimer(image):
33
  result = alzheimer_classifier(image)
 
34
  prediction = {}
35
  for ele in result:
36
+ label, score = ele.values()
37
  prediction[label] = score
38
  return prediction
39