Update app.py

app.py

@@ -20,12 +20,14 @@ with gr.Blocks() as demo:
     """)
 
     # Model Evaluation
-    gr.Interface(
-        fn=predict,
-        inputs=gr.Image(type="pil"),
-        outputs=[gr.Image(type="pil", label="Image Prediction"),
-                 gr.Textbox(type="text", label="Price Prediction")]
-    )
+    # gr.Interface(
+    #     fn=predict,
+    #     inputs=gr.Image(type="pil"),
+    #     outputs=[gr.Image(type="pil", label="Image Prediction"),
+    #              gr.Textbox(type="text", label="Price Prediction")]
+    # )
+    gr.Interface(inputs=["image"],outputs=["image"],fn=lambda img:model(img).render()[0]).launch()
+
 
 if __name__ == "__main__":
     demo.launch()