h

app.py

@@ -54,13 +54,15 @@ def process_video(video_path):
     
     return output_animation
 
-def predict(input):
-    if isinstance(input, str):  # video 
-        return process_video(input)
-    else:  # image
-        bundle = make_bundle(input, "Model3D")
+def predict(input_img, input_video):
+    if input_video:
+        return process_video(input_video)
+    elif input_img is not None:
+        bundle = make_bundle(input_img, "Model3D")
         dataset = EvalWMetaDataset(opts, [bundle])
         return reconstructor.evaluate(dataset)
+    else:
+        return None, None
 
 footer = r"""
 <center>
@@ -75,7 +77,7 @@ with gr.Blocks(title="PIFuHD") as app:
     with gr.Row(equal_height=False):
         with gr.Column():
             input_img = gr.Image(type="numpy", label="Input image")
-            input_video = gr.Video(type="filepath", label="Input Video")
+            input_video = gr.Video(label="Input Video")
             run_btn = gr.Button(variant="primary")
         with gr.Column():
             output_obj = gr.Model3D(label="Output model")