Update model/model.py

model/model.py

@@ -17,4 +17,4 @@ def predict(image_path):
     labels = ['Glioma', 'Meningioma', 'No tumor', 'Pituitary']
     probabilities = {label: probability for label, probability in zip(labels, prediction[0])}
     output_text = "\n".join([f"{label}: {probability:.2f}" for label, probability in probabilities.items()])
-    return output_text
+    return probabilities