default and safe

app.py

@@ -50,26 +50,13 @@ def infer(prompt, n_samples, steps, scale, seed):
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)
 
-    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"
-    })
+    url = os.getenv('BACKEND_URL_SAFE')
+    response = requests.get(url.format(prompt, int(n_samples), max(50, int(steps)), f'{scale:.1f}', int(seed)))
 
     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.")
+        raise gr.Error("An error occurred.")
     else:
-        if data['output']['result_type'] == "error":
-            raise gr.Error(data['output']['value'])
         for image in data['output']['choices']:
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)