new endpoint

app.py

@@ -18,6 +18,20 @@ 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_SAFE_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,
+    }, headers={
+        "User-Agent": "hfdemo"
+    })
+    return res
 
 def infer(prompt, n_samples, steps, scale, seed):
     global is_gpu_busy
@@ -39,7 +53,7 @@ def infer(prompt, n_samples, steps, scale, seed):
         "seed": seed,
         "guidance_scale": scale,
     }, headers={
-        "User-Agent": "YOUR_NAME_HERE"
+        "User-Agent": "hfdemo"
     })
 
     data = json.load(BytesIO(response.content))
@@ -52,8 +66,7 @@ def infer(prompt, n_samples, steps, scale, seed):
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)
 
-    url = os.getenv('BACKEND_URL_SAFE')
-    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)), scale, seed)
 
     data = json.load(BytesIO(response.content))
     if 'output' not in data:
@@ -211,7 +224,7 @@ examples = [
         1,
         50,
         7.5,
-        557645701,
+        557645701
     ],
     [
         'portrait of girl with smokey eyes makeup in abandoned hotel, grange clothes, redshift, wide high angle coloured polaroid photograph with flash, kodak film, hyper real, stunning moody cinematography, with anamorphic lenses, by maripol, fallen angels by wong kar - wai, style of suspiria and neon demon and children from bahnhof zoo, detailed ',