updated app

app.py

@@ -198,6 +198,11 @@ def video_inference(
     
     return "avatar.gif"
 
+upload = True
+
+def toggle_webcam():
+    upload = not upload
+
 description = """<p style='text-align: center'> Create a personal avatar from just a single image using ROME. <br> <a href='https://arxiv.org/abs/2206.08343' target='_blank'>Paper</a> | <a href='https://samsunglabs.github.io/rome' target='_blank'>Project Page</a> | <a href='https://github.com/SamsungLabs/rome' target='_blank'>Github</a> </p>"""
 quote = """
 > <p style='text-align: center'> [The] system creates realistic mesh-based avatars from a single <strong>source</strong> photo. These avatars are rigged, i.e., they can be driven by the animation parameters from a different <strong>driving</strong> frame. </p>"""
@@ -217,9 +222,10 @@ with gr.Blocks() as demo:
     with gr.Tab("Video Inference"):
         with gr.Row():
             source_img2 = gr.Image(type="pil", label="Source image", show_label=True)
-            driver_vid = gr.Video(label="Driver video")
+            driver_vid = gr.Video(label="Driver video", source= "upload" if upload else "webcam" )
         video_output = gr.Image(label="Rendered GIF avatar")
         video_button = gr.Button("Predict")
+        toggle_cam = gr.Button("Toggle Webcam")
 
     gr.Examples(
         examples=[
@@ -234,5 +240,6 @@ with gr.Blocks() as demo:
 
     image_button.click(image_inference, inputs=[source_img, driver_img], outputs=image_output)
     video_button.click(video_inference, inputs=[source_img2, driver_vid], outputs=video_output)
+    toggle_cam.click(toggle_webcam)
 
 demo.launch()