Update app.py

app.py

@@ -78,20 +78,19 @@ def generate_image(prompt):
     """Generates an image using the refined prompt."""
     # truncated_prompt = truncate_prompt(prompt)
     try:
-        # image = pipe(
-        # truncated_prompt,
-        # height=1024,
-        # width=1024,
-        # guidance_scale=3.5,
-        # num_inference_steps=50,
-        # max_sequence_length=512,
-        # generator=torch.Generator("cpu").manual_seed(0)
-        image = pipe(truncated_prompt).images[0]
-# ).images[0]
+        image = pipe(
+        prompt,
+        height=1024,
+        width=1024,
+        guidance_scale=3.5,
+        num_inference_steps=50,
+        max_sequence_length=512,
+        generator=torch.Generator("cpu").manual_seed(0)
+).images[0]
     except Exception as e:
         # Optionally, handle errors (you can also return a default error image)
         image = None
-        print("No image")
+        print(e)
     return image
 
 with gr.Blocks(css=".gradio-container {background-color: #f9f9f9; padding: 20px;}") as demo: