Update app.py

app.py

@@ -314,22 +314,38 @@ with gr.Blocks(
         gr.HTML(gallery_html)
 
 # 이벤트 핸들러 부분만 수정
-    def generate_image(*args):
-        return process_and_save_image(*args)
+    def generate_image(h, w, s, sc, p, sd):
+        return process_and_save_image(
+            height=h,
+            width=w,
+            steps=s,
+            scales=sc,
+            prompt=p,
+            seed=sd
+        )
 
-    def update_seed():
-        return get_random_seed()
+    def update_random_seed():
+        return gr.Number.update(value=get_random_seed())
 
+    # 이벤트 바인딩
     generate_btn.click(
-        fn=generate_image,
-        inputs=[height, width, steps, scales, prompt, seed],
+        generate_image,
+        inputs={
+            "h": height,
+            "w": width,
+            "s": steps,
+            "sc": scales,
+            "p": prompt,
+            "sd": seed
+        },
         outputs=output
     )
 
     randomize_seed.click(
-        fn=update_seed,
+        update_random_seed,
+        inputs=None,
         outputs=seed
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR], share=True)
+    demo.launch(allowed_paths=[PERSISTENT_DIR])