Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import numpy as np
 import cv2
 import tensorflow as tf
-import matplotlib
-matplotlib.use('Agg')
 
 # Load the model
 model = tf.keras.models.load_model('model.h5')
@@ -26,7 +24,7 @@ def predict_pneumonia(image):
         pneumonia_prediction = 0
 
     # Return the probability of each class
-    class_probabilities = model.predict(img_array)
+    class_probabilities = np.array(model.predict(img_array))
 
     # Return the top two possible classifications
     top_classes = np.argsort(class_probabilities[0])[-2:]