updated endpoint to universal

app.py

@@ -18,13 +18,13 @@ 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):
+def safe_sd(prompt, n_samples, steps, scale, seed, mode):
     url = os.getenv('BACKEND_URL_SAFE_NEW')
     res = requests.post(url, json={
         "model": "together/universal-sd",
         "prompt": prompt,
         "n": n_samples,
-        "mode": "safe_text2img",
+        "mode": mode,
         "steps": steps,
         "seed": seed,
         "guidance_scale": scale,
@@ -41,20 +41,8 @@ def infer(prompt, n_samples, steps, scale, seed):
 
     if prompt == "":
         raise gr.Error("Empty prompt. Please provide a prompt.")
-    url = os.getenv('BACKEND_URL_DEFAULT')
 
-    response = requests.post(url, json={
-        "model": "StableDiffusion",
-        "prompt": prompt,
-        "n": int(n_samples),
-        "height": 512,
-        "width": 512,
-        "steps": max(50,int(steps)),
-        "seed": seed,
-        "guidance_scale": scale,
-    }, headers={
-        "User-Agent": "hfdemo"
-    })
+    response = safe_sd(prompt, int(n_samples), max(50,int(steps)), scale, seed, mode="text2img")
 
     data = json.load(BytesIO(response.content))
     if 'output' not in data:
@@ -66,7 +54,7 @@ def infer(prompt, n_samples, steps, scale, seed):
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)
 
-    response = safe_sd(prompt, int(n_samples), max(50,int(steps)), scale, seed)
+    response = safe_sd(prompt, int(n_samples), max(50,int(steps)), scale, seed, mode="safe_text2img")
 
     data = json.load(BytesIO(response.content))
     if 'output' not in data: