syntax

app.py

@@ -22,6 +22,7 @@ def generate_image(lora_scale, guidance_scale, prompt_strength, num_steps, promp
         }
     )
     prediction.wait()
+    output = prediction.output
     image_url = output[0] if output else None
     return image_url