File size: 890 Bytes
e547b24
 
 
1ecc2a8
4815fd5
 
e547b24
1ecc2a8
e547b24
 
1ecc2a8
e547b24
1ecc2a8
 
 
 
 
e547b24
1ecc2a8
 
e547b24
1ecc2a8
 
 
e547b24
1ecc2a8
e547b24
1ecc2a8
e547b24
1ecc2a8
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
import gradio as gr
import requests
from PIL import Image
import io
import os 


API_URL = "https://api-inference.huggingface.co/models/Blane187/kana-arima-s1-ponyxl-lora-nochekaiser"
API_TOKEN = os.getenv("HF_READ_TOKEN")

headers = {"Authorization": f"Bearer {API_TOKEN}"}

def query(inputs):
    response = requests.post(API_URL, headers=headers, json={"inputs": inputs})
    image_bytes = response.content
    image = Image.open(io.BytesIO(image_bytes))
    return image

with gr.Blocks() as demo:
    gr.Markdown("## Generate an Image using Hugging Face Model")
    
    with gr.Row():
        prompt_input = gr.Textbox(label="Enter a prompt", placeholder="Astronaut riding a horse")
        generate_btn = gr.Button("Generate Image")
    
    output_image = gr.Image(label="Generated Image")

    generate_btn.click(fn=query, inputs=prompt_input, outputs=output_image)

demo.launch()