Image to refer test

app.py

@@ -10,9 +10,6 @@ from huggingface_hub import snapshot_download
 SPACE_USERNAME = 'KR_4dmin'
 SPACE_PASSWORD = 'KR_4dmin'
 
-if not SPACE_USERNAME or not SPACE_PASSWORD:
-    raise ValueError("Las credenciales no están configuradas correctamente.")
-
 snapshot_download(repo_id="Roomie/xavyy", cache_dir='./')
 
 pipeline = AutoPipelineForText2Image.from_pretrained(
@@ -25,7 +22,7 @@ MAX_IMAGE_SIZE = 1024
 
 
 @spaces.GPU
-def infer(prompt, height, width):
+def infer(prompt, refer_image, height, width):
 
     image = pipeline(
         prompt=prompt,
@@ -33,10 +30,11 @@ def infer(prompt, height, width):
         # num_inference_steps=num_inference_steps,
         width=width,
         height=height,
+        image=refer_image
         # generator=generator
     ).images[0]
 
-    return image
+    return 'image'
 
 
 examples = [
@@ -68,9 +66,11 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
 
-            run_button = gr.Button("Run", scale=0)
+            refer_image = gr.Image(show_label=True, label="Reference image")
+
+        run_button = gr.Button("Run", scale=0)
 
-        result = gr.Image(label="Result", show_label=False)
+        result = gr.Image(label="Result", show_label=True)
 
         with gr.Accordion("Advanced Settings", open=False):
 
@@ -138,7 +138,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
-        inputs=[prompt, height, width],
+        inputs=[prompt, refer_image, height, width],
         outputs=[result]
     )