Modify image generation function return values and Gradio interface

app.py

@@ -138,9 +138,9 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
 
         torch.cuda.empty_cache()  # Clean up GPU memory after generation
         progress(1.0, desc="Done!")
-        return prompt, image, seed
+        return image, seed
     except Exception as e:
-        return None, None, f"Error generating image: {str(e)}"
+        return None, f"Error generating image: {str(e)}"
  
 examples = [
     "a tiny astronaut hatching from an egg on the moon",
@@ -332,7 +332,7 @@ def create_interface():
             triggers=[run_button.click],
             fn = infer,
             inputs = [refined_prompt, seed, randomize_seed, width, height, num_inference_steps],
-            outputs = [prompt, generated_image, seed]
+            outputs = [generated_image, prompt]
         )
 
     return demo