Upload app.py

app.py

@@ -0,0 +1,40 @@
+import gradio as gr
+import requests
+import base64
+from PIL import Image
+from io import BytesIO
+
+API_URL = "https://huggingface.co/spaces/florafeng/UCLIP-IMG-GEN/"
+
+
+def generate(prompt, seed):
+    response = requests.post(
+        API_URL,
+        json={"prompt": prompt, "seed": int(seed)}
+    )
+
+    data = response.json()
+
+    img_bytes = base64.b64decode(data["image"])
+    image = Image.open(BytesIO(img_bytes))
+
+    return image
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎨 AI Bedroom Generator")
+
+    prompt = gr.Textbox(
+        label="Prompt",
+        value="a unique bedroom with dolls, pillows, paintings"
+    )
+
+    seed = gr.Number(value=123, label="Seed")
+
+    btn = gr.Button("Generate")
+
+    output = gr.Image()
+
+    btn.click(generate, inputs=[prompt, seed], outputs=output)
+
+demo.launch()