Update app.py

app.py

@@ -317,7 +317,11 @@ with gr.Blocks(theme='gradio/soft') as demo:
         with gr.Column():
             prompt_input = gr.Textbox(label="Prompt")
             negative_prompt = gr.Textbox(label="Negative Prompt")
-            input_image = gr.Image(label="Input Image")
+            with gr.Box():
+                with gr.Tab("Upload Image"):
+                    upload_image = gr.Image(label="Upload Image", source="upload")
+                with gr.Tab("Webcam"):
+                    webcam_image = gr.Image(label="Webcam", source="webcam")
             # output_image = gr.Gallery(label='Output Image', show_label=False, elem_id="gallery").style(grid=3, height='auto')
             submit_btn = gr.Button(value = "Submit")
             # inputs = [prompt_input, negative_prompt, input_image]
@@ -354,12 +358,17 @@ with gr.Blocks(theme='gradio/soft') as demo:
                "example4.png"
             ],
         ],
-        inputs=[prompt_input, negative_prompt, input_image, model_type],
+        inputs=[prompt_input, negative_prompt, upload_image, model_type],
         outputs=[output_image, system_info],
         fn=infer,
         cache_examples=True,
     )
-            
+    # check source of image
+    if upload_image and webcam_image is None:
+        input_image = upload_image
+    else:
+        input_image = webcam_image
+        
     inputs = [prompt_input, negative_prompt, input_image, model_type]
     submit_btn.click(fn=infer, inputs=inputs, outputs=[output_image, system_info])