Update app.py

app.py

@@ -18,8 +18,9 @@ def funksjon(input_img):
     
     predictions = model.predict(input_img)
     #imgPred = input_img
-    
-    return predictions
+    pil_image = Image.fromarray(predictions)
+    # Return the image
+    return pil_image
 
 
 demo = gr.Interface(funksjon, gr.Image(), "image")