Update app.py

app.py

@@ -1,4 +1,16 @@
 import gradio as gr
+import random
+import requests
+import io
+from PIL import Image
+
+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):
     return None
@@ -15,7 +27,7 @@ theme = gr.themes.Base(
 )
 
 with gr.Blocks(theme=theme) as demo:
-    img = gr.Image(show_label=False)
+    img = gr.Image(show_label=False, type='pil')
     textbox = gr.Textbox(show_label=False)
     button = gr.Button("generate")
     button.click(fn=generate, inputs=textbox, outputs=img)