Update app.py

app.py

@@ -230,8 +230,8 @@ class Model:
         video.close()
 
 
-def gen_fn(use_seed, seed, psi):
-    z = RandomState(int(seed) + 2 ** 31).randn(1, 512) if use_seed else np.random.randn(1, 512)
+def gen_fn(method, seed, psi):
+    z = RandomState(int(seed) + 2 ** 31).randn(1, 512) if method == 1 else np.random.randn(1, 512)
     w = model.get_w(z.astype(dtype=np.float32), psi)
     img_out = model.get_img(w)
     return img_out, json.dumps(w.tolist()), img_out
@@ -247,7 +247,7 @@ def encode_img_fn(img):
 
 
 def gen_video_fn(w1, w2, frame):
-    if w1 is None or w2 is None:
+    if w1 is None or w2 is None or w1 == "" or w2 == "":
         return None
     model.gen_video(np.array(json.loads(w1), dtype=np.float32), np.array(json.loads(w2), dtype=np.float32), "video.mp4",
                     int(frame))
@@ -266,7 +266,7 @@ if __name__ == '__main__':
                 with gr.Row():
                     with gr.Column():
                         gr.Markdown("generate image randomly or by seed")
-                        gen_input1 = gr.Checkbox(value=False, label="use seed")
+                        gen_input1 = gr.Radio(label="method", choices=["random", "use seed"], type="index")
                         gen_input2 = gr.Number(value=1, label="seed")
                         gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.6, label="truncation psi")
                         with gr.Group():