app.py

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',
)

with gr.Blocks(theme=theme) as demo:
 img = gr.Image(show_label=False, type='pil', height=512)
 textbox = gr.Textbox(show_label=False)
 button = gr.Button("generate", variant="primary")
 button.click(fn=generate, inputs=textbox, outputs=img)

demo.launch()