post

app.py

@@ -27,12 +27,27 @@ def infer(prompt, n_samples, steps, scale, seed):
     images_default = []
 
     url = os.getenv('BACKEND_URL_DEFAULT')
-    response = requests.get(url.format(prompt, int(n_samples), max(50,int(steps)), f'{scale:.1f}', int(seed)))
+
+    response = requests.post(url, json={
+        "model": "StableDiffusion",
+        "prompt": prompt,
+        "n": int(n_samples),
+        "height": 512,
+        "width": 512,
+        "steps": max(50,int(steps)),
+        "seed": int(seed),
+        "guidance_scale": scale,
+    }, headers={
+        "User-Agent": "YOUR_NAME_HERE"
+    })
+
+    #response = requests.get(url.format(prompt, int(n_samples), max(50,int(steps)), f'{scale:.1f}', int(seed)))
     #response = requests.get(url.format('a%20naked%20girl', 2, 50, 7.5, 2))
     data = json.load(BytesIO(response.content))
     if 'output' not in data:
         raise gr.Error("Although safety guidance is enabled, potential unsafe content found. Please try again with different seed.")
     else:
+        if data['output']['result_type']
         for image in data['output']['choices']:
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)