import gradio as gr import random import requests import io from PIL import Image import os API_URL = "https://api-inference.huggingface.co/models/tonyassi/tony-assi-lora-1" headers = {"Authorization": "Bearer " + os.environ.get('TOKEN')} def query(payload): response = requests.post(API_URL, headers=headers, json=payload) return response.content def generate(prompt): image_bytes = query({ "inputs": "Tony Assi style " + prompt, "parameters" : { "negative_prompt": "ugly, deformed, bad quality", "seed": random.randint(0,9999999)} }) image = Image.open(io.BytesIO(image_bytes)) return image theme = gr.themes.Base( primary_hue="gray", secondary_hue="gray", neutral_hue="gray", font=['Helvetica', 'ui-sans-serif', 'system-ui', 'sans-serif'], ).set( button_large_text_weight='400', input_background_fill='#ffffff', #input_border_width='*block_border_width', #button_primary_background_fill='#ffffff', #button_border_width='*block_border_width', ) with gr.Blocks(theme=theme) as demo: img = gr.Image(show_label=False, type='pil') textbox = gr.Textbox(show_label=False, placeholder='type your prompt in here') button = gr.Button("generate", variant="primary") button.click(fn=generate, inputs=textbox, outputs=img) textbox.submit(fn=generate, inputs=textbox, outputs=img) demo.launch()