Create app.py

app.py

@@ -0,0 +1,44 @@
+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()