florafeng's picture
Upload app.py
a02ac6f verified
raw
history blame contribute delete
920 Bytes
import gradio as gr
from gradio_client import Client
# 🔴 Replace with your backend Space URL
BACKEND_URL = "https://florafeng-uclip-img-gen.hf.space"
client = Client(BACKEND_URL)
def generate(prompt, seed):
try:
result = client.predict(
prompt,
seed,
api_name="/predict" # default for gr.Interface
)
return result
except Exception as e:
return f"Error: {str(e)}"
with gr.Blocks() as demo:
gr.Markdown("# 🎨 AI Bedroom Generator")
with gr.Row():
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(label="Generated Image")
btn.click(
fn=generate,
inputs=[prompt, seed],
outputs=output
)
demo.launch()