Update app.py

app.py

@@ -37,7 +37,14 @@ def predict_image(image):
     return label
 
 # Gradio interface
-with gr.Interface(fn=predict_image, 
-                  inputs=gr.inputs.Image(type="bytes"), 
-                  outputs=gr.outputs.Textbox()) as demo:
-    demo.launch(debug=True)
+with gr.Blocks() as demo:
+    gr.Markdown("### Image Classification with MobileNetV2")
+    
+    with gr.Row():
+        image_input = gr.Image(type="bytes", label="Upload Image")
+        result_output = gr.Textbox(label="Predicted Label")
+        
+    # Endpoint for inference
+    gr.Interface(fn=predict_image, inputs=image_input, outputs=result_output, api_name="/predict_image")
+
+demo.launch(debug=True)