Update app.py

app.py

@@ -47,10 +47,10 @@ except Exception as e:
     print(f"Failed to load the model: {e}")
 
 # 이미지 생성 함수
-@spaces.GPU(duration=120)
 def generate_image(prompt, width=1024, height=1024, scales=5, steps=4, seed=-1, nums=1, progress=gr.Progress(track_tqdm=True)):
     if pipe is None:
-        return "Model is not loaded properly. Please check the logs for details.", None
+        print("Model is not loaded properly. Please check the logs for details.")
+        return None, "Model not loaded."
 
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
@@ -59,12 +59,12 @@ def generate_image(prompt, width=1024, height=1024, scales=5, steps=4, seed=-1,
     generator = torch.Generator().manual_seed(seed)
 
     try:
-        image = pipe(prompt=text, height=height, width=width, guidance_scale=scales, output_type="pil", num_inference_steps=steps, max_sequence_length=512, num_images_per_prompt=nums, generator=generator).images
+        images = pipe(prompt=text, height=height, width=width, guidance_scale=scales, num_inference_steps=steps, max_sequence_length=512, num_images_per_prompt=nums, generator=generator).images
     except Exception as e:
         print(f"Error generating image: {e}")
-        return None, seed
+        return None, "Error during image generation."
 
-    return image, seed
+    return images, seed
 
 # Gradio 인터페이스 구성 및 실행
 with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
@@ -86,3 +86,4 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     prompt.submit(fn=generate_image, inputs=[prompt, width, height, scales, steps, seed, nums], outputs=[img, seed])
 
 demo.queue().launch()
+