Spaces:

RoCobo
/

WiggleGAN

Runtime error

Rodrigo_Cobo commited on Aug 22, 2022

Commit

4edfa31

•

1 Parent(s): 82346e7

fix depth output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def calculate_depth(model_type, gan_type, dim, slider, img):
     path_video = os.path.join(os.path.dirname(__file__), 'WiggleResults' , name_output + '_0.mp4')
     print(path_video)
-    return [img,f'WiggleResults/' + name_output + '_0.gif', path_video, f'WiggleResults/'+ name_output + '.jpg']
 with gr.Blocks() as demo:

     path_video = os.path.join(os.path.dirname(__file__), 'WiggleResults' , name_output + '_0.mp4')
     print(path_video)
+    return [out_im,f'WiggleResults/' + name_output + '_0.gif', path_video, f'WiggleResults/'+ name_output + '.jpg']
 with gr.Blocks() as demo: