Update app.py

app.py

@@ -33,9 +33,8 @@ def classify_image(input_image):
     return img2  # Return the image with prediction annotations
 
 # Interface creation using Gradio
-inputs = gr.inputs.Image()
-outputs = gr.outputs.Image()
-interface = gr.Interface(classify_image, inputs, outputs, capture_session=True)
+inputs = gr.Image()
+interface = gr.Interface(classify_image, inputs, capture_session=True)
 
 # Launch the Gradio app
 interface.launch()