Update app.py

app.py

@@ -32,6 +32,17 @@ def infer(prompt):
         images.append(image_b64)
     
     return images
+
+    def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+
+output = query({
+	"inputs": "Astronaut riding a horse",
+	"parameters": {
+		"num_inference_steps": 25,
+		"negative_prompt": "deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, (NSFW:1.25)",
+		"num_images_per_prompt": 4
     
     
 css = """