Update app.py

app.py

@@ -187,7 +187,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
             try:
                 generated_image = pipe(
-                    prompt=[formatted_prompt],  # 수정된 프롬프트 사용
+                    prompt=[formatted_prompt],
                     generator=torch.Generator().manual_seed(int(seed)),
                     num_inference_steps=int(steps),
                     guidance_scale=float(scales),
@@ -200,31 +200,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 if saved_path is None:
                     print("Warning: Failed to save generated image")
                 
-                return generated_image, load_gallery()
+                return generated_image
             except Exception as e:
                 print(f"Error in image generation: {str(e)}")
-                return None, load_gallery()
-    
-
+                return None
     
     def update_seed():
         return get_random_seed()
 
     generate_btn.click(
         process_and_save_image,
-        inputs=[height, width, steps, scales, prompt, seed],
-        outputs=[output, gallery]
-    )
-    
-    randomize_seed.click(
-        update_seed,
-        outputs=[seed]
-    )
-    
-    generate_btn.click(
-        update_seed,
-        outputs=[seed]
-    )
-
-if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR])
+        inputs=[height, width, steps, scales, prompt, seed],