Fix output type

main.py

@@ -28,7 +28,7 @@ def run():
     demo = gr.Interface(
         fn=predict,
         inputs=gr.components.Image(),
-        outputs=gr.components.Textbox(),
+        outputs=gr.components.Image(),
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)