Update app.py

app.py

@@ -5,20 +5,28 @@ 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')}
+
+from huggingface_hub import InferenceClient
+client = InferenceClient("tonyassi/tony-assi-lora-1", token=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))
+    """
+    
+    # output is a PIL.Image object
+    image = client.text_to_image(prompt)
     return image
 
 theme = gr.themes.Base(