Update app.py

app.py

@@ -29,11 +29,11 @@ def predict_pneumonia(image):
     class_probabilities = model.predict(img_array)
 
     # Return the top two possible classifications
-    top_classes = model.predict_classes(img_array)
+    top_classes = np.argsort(class_probabilities[0])[-2:]
 
     # Return the class names and the confidence scores for each class
     class_names = ["Pneumonia", "Normal"]
-    confidence_scores = class_probabilities
+    confidence_scores = class_probabilities[0]
 
     return {
         "Pneumonia": pneumonia_prediction,
@@ -59,8 +59,7 @@ gradio_interface = gr.Interface(
         ["NORMAL2-IM-1431-0001.jpeg"]
     ],
     theme="default",
-    allow_uploads=True
+    allow_flagging=False
 )
 
-
 gradio_interface.launch()