Update app.py

app.py

@@ -21,21 +21,14 @@ def predict(image,max_length=64, num_beams=4):
   return caption_text 
 
 
+# Gradio Interface
+gradio_app = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(label="Select image for captioning", sources=['upload', 'webcam'], type="pil"),
+    outputs=[gr.Image(label="Processed Image"), gr.Textbox(label="Image Caption")],
+    examples = [f"example{i}.jpg" for i in range(1,7)],
+    title="Image Captioning with our model",
+)
 
-input = gr.Image(label="Upload any Image", type = 'pil', optional=True)
-output = [gr.Image(label="Processed Image"), gr.Textbox(label="Image Caption")]
-examples = [f"example{i}.jpg" for i in range(1,7)]
-
-title = "Image Captioning "
-description = "Made by : shreyasdixit.tech"
-interface = gr.Interface(
-            
-        fn=predict,
-        description=description,
-        inputs = input,
-        theme="grass",
-        outputs=output,
-        examples = examples,
-        title=title,
-    )
-interface.launch(debug=True)
+if __name__ == "__main__":
+    gradio_app.launch()