Update app.py

app.py

@@ -22,12 +22,9 @@ model_path = 'densenet-9.onnx'
 model = onnx.load(model_path)
 session = ort.InferenceSession(model.SerializeToString())
 
-def get_image(path, show=False):
+def get_image(path):
     with Image.open(path) as img:
         img = np.array(img.convert('RGB'))
-    if show:
-        plt.imshow(img)
-        plt.axis('off')
     return img
     
 def preprocess(img):
@@ -44,7 +41,7 @@ def preprocess(img):
     return img
 
 def predict(path):
-    img = get_image(path, show=True)
+    img = get_image(path)
     img = preprocess(img)
     ort_inputs = {session.get_inputs()[0].name: img}
     preds = session.run(None, ort_inputs)[0]