Update app.py

app.py

@@ -95,4 +95,19 @@ def generate_image(uploaded_image, prompt):
             image_embeds=face_emb,
             num_inference_steps=20,
             guidance_scale=7.5,
-            height=512,
+            height=512,
+            width=512,
+        ).images[0]  # Closing parenthesis here
+        return "Image generated successfully!", image
+    except Exception as e:
+        return f"Generation failed: {e}", None
+
+interface = gr.Interface(
+    fn=generate_image,
+    inputs=[gr.Image(type="pil", label="Upload Reference Image"), gr.Textbox(label="Enter Prompt", placeholder="e.g., A photorealistic astronaut in space")],
+    outputs=[gr.Textbox(label="Status"), gr.Image(label="Generated Image")],
+    title="Face Reference Image Generator (Kolors with IP-Adapter)",
+    description="Upload an image with a face, enter a prompt, and generate a new image preserving the reference face."
+)
+
+interface.launch()