Update app.py

app.py

@@ -11,9 +11,14 @@ def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.content
 
-
 def generate(prompt):
-    return None
+    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",