new endpoint

app.py

@@ -18,6 +18,18 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 
 is_gpu_busy = False
 
+def safe_sd(prompt, n_samples, steps, scale, seed):
+    url = os.getenv('BACKEND_URL_NEW')
+    res = requests.post(url, json={
+        "model": "together/universal-sd",
+        "prompt": prompt,
+        "n": n_samples,
+        "mode": "safe_text2img",
+        "steps": steps,
+        "seed": seed,
+        "guidance_scale": scale,
+    })
+    return res
 
 def infer(prompt, n_samples, steps, scale, seed):
     global is_gpu_busy
@@ -41,9 +53,11 @@ def infer(prompt, n_samples, steps, scale, seed):
     #           images.append(image)
     # else:
     url = os.getenv('BACKEND_URL')
-    response = requests.get(url.format(prompt, int(n_samples), max(50,int(steps)), f'{scale:.1f}', int(seed)))
+    response = safe_sd(prompt, int(n_samples), max(50,int(steps)), f'{scale:.1f}', int(seed))
+    #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))
+    #data = json.load(BytesIO(response.content))
+    data = response.json()
     if 'output' not in data:
         raise gr.Error("Although safety guidance is enabled, potential unsafe content found. Please try again with different seed.")
     else: