Update app.py

app.py

@@ -87,6 +87,7 @@ def predict(classification_mode, image):
 
   labels = model.dls.vocab
   # Classification model prediction
+  image  = PILImage.create(image)
   pred, pred_idx, probs = model.predict(image)
 
   seg_img = None
@@ -132,7 +133,7 @@ description = """
 gr.Interface(fn=predict,
              inputs= [gr.Dropdown(choices=['Binary Classification', 'Multiclass Classification'], label='Classification Mode:', 
                                   info='Choose to classify between anomaly and no-anomaly OR between 12 different types of anomalies.'),
-                      gr.Image(type='pil', label='Input infrared image: ')],
+                      gr.Image(label='Input infrared image: ')],
              outputs=[gr.outputs.Label(num_top_classes=3, label='Detected:').style(container=False),
                       gr.Image(type='pil', label=' ').style(height=240, width=144),
                       gr.Textbox(label='Affected area:').style(container=False)],