Update app.py

app.py

@@ -27,7 +27,10 @@ def classify_alzheimer(image):
     binary_data = buffer.tobytes()
 
     response = requests.post(ALZHEIMER_API_URL, headers=headers, data=binary_data)
-    result = {item['label']: item['score'] for item in response.json()}
+    result = {}
+    for ele in response:
+        label, score = ele.values()
+        result[label] = score
 
     return result
 
@@ -39,7 +42,10 @@ def classify_breast_cancer(image):
     binary_data = buffer.tobytes()
 
     response = requests.post(BREAST_CANCER_API_URL, headers=headers, data=binary_data)
-    result = {item['label']: item['score'] for item in response.json()}
+    result = {}
+    for ele in response:
+        label, score = ele.values()
+        result[label] = score
 
     return result
 
@@ -51,7 +57,10 @@ def classify_brain_tumor(image):
     binary_data = buffer.tobytes()
 
     response = requests.post(BRAIN_TUMOR_API_URL, headers=headers, data=binary_data)
-    result = {item['label']: item['score'] for item in response.json()}
+    result = {}
+    for ele in response:
+        label, score = ele.values()
+        result[label] = score
 
     return result