v2

app.py

@@ -17,7 +17,7 @@ from infer import lotus, lotus_video
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-def infer(path_input, seed=0):
+def infer(path_input, seed):
     name_base, name_ext = os.path.splitext(os.path.basename(path_input))
     output_g, output_d = lotus(path_input, 'normal', seed, device)
     if not os.path.exists("files/output"):
@@ -28,7 +28,7 @@ def infer(path_input, seed=0):
     output_d.save(d_save_path)
     return [path_input, g_save_path], [path_input, d_save_path]
 
-def infer_video(path_input, seed=0):
+def infer_video(path_input, seed):
     frames_g, frames_d = lotus_video(path_input, 'normal', seed, device)
     if not os.path.exists("files/output"):
         os.makedirs("files/output")
@@ -142,10 +142,10 @@ def run_demo_server():
                 gr.Examples(
                     fn=infer,
                     examples=sorted([
-                        os.path.join("files", "images", name)
+                        [os.path.join("files", "images", name), 0]
                         for name in os.listdir(os.path.join("files", "images"))
                     ]),
-                    inputs=[image_input],
+                    inputs=[image_input, seed],
                     outputs=[image_output_g, image_output_d],
                     cache_examples=True,
                 )
@@ -188,10 +188,10 @@ def run_demo_server():
                 gr.Examples(
                     fn=infer_video,
                     examples=sorted([
-                        os.path.join("files", "videos", name)
+                        [os.path.join("files", "videos", name), 0]
                         for name in os.listdir(os.path.join("files", "videos"))
                     ]),
-                    inputs=[input_video],
+                    inputs=[input_video, seed],
                     outputs=[video_output_g, video_output_d],
                     cache_examples=True,
                 )