Update app.py
Browse files
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 |
-
|
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 |
|