File size: 920 Bytes
9cad172
a02ac6f
9cad172
a02ac6f
 
9cad172
a02ac6f
9cad172
7f23a9a
a02ac6f
 
 
 
 
 
 
e12fb24
9cad172
a02ac6f
 
9cad172
 
 
 
 
a02ac6f
 
 
 
 
 
9cad172
 
 
a02ac6f
 
 
 
 
 
 
9cad172
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
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()