Update main.py

main.py

@@ -46,11 +46,15 @@ def predict(inp):
 
 
 def run():
-    demo = gr.Interface(
-        fn=predict,
-        inputs=gr.inputs.Image(type="pil"),
-        outputs=gr.outputs.Label(num_top_classes=3),
-    )
+    # demo = gr.Interface(
+    #     fn=predict,
+    #     inputs=gr.inputs.Image(type="pil"),
+    #     outputs=gr.outputs.Label(num_top_classes=3),
+    # )
+
+    demo = gr.Interface(fn=infer, 
+         inputs=['image'],
+         outputs=['image', 'text'])
 
     demo.launch(server_name="0.0.0.0", server_port=7860)