Update app.py

app.py

@@ -13,9 +13,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
 # Captioning function
-def generate_caption(upload_img, webcam_img):
+def generate_caption(image):
     # Choose image from upload or webcam
-    image = webcam_img if webcam_img is not None else upload_img
     if image is None:
         return "No image provided."
     # Preprocess
@@ -29,13 +28,12 @@ def generate_caption(upload_img, webcam_img):
 with gr.Blocks() as demo:
     gr.Markdown("# Image Captioning with Gradio")
     with gr.Row():
-        upload_input = gr.Image(source="upload", type="pil", label="Upload Image")
-        webcam_input = gr.Image(source="webcam", type="pil", label="Use Camera")
+        upload_input = gr.Image(sources=["upload", "webcam", "clipboard"], type="pil", label="Upload Image")
     output_text = gr.Textbox(label="Caption", interactive=False)
     generate_btn = gr.Button("Generate Caption")
     generate_btn.click(
         fn=generate_caption,
-        inputs=[upload_input, webcam_input],
+        inputs=upload_input,
         outputs=output_text
     )