Update app.py

app.py

@@ -20,8 +20,10 @@ is_gpu_busy = False
 
 def safe_sd(prompt, n_samples, steps, scale, seed, mode):
     url = os.getenv('BACKEND_URL_SAFE_NEW')
+    token = os.getenv('BACKEND_TOKEN')
+    user = os.getenv('BACKEND_USER')
     res = requests.post(url, json={
-        "model": "together/universal-sd",
+        "model": "togethercomputer/UniversalSD",
         "prompt": prompt,
         "n": n_samples,
         "mode": mode,
@@ -29,7 +31,8 @@ def safe_sd(prompt, n_samples, steps, scale, seed, mode):
         "seed": seed,
         "guidance_scale": scale,
     }, headers={
-        "User-Agent": "hfdemo"
+        "Authorization": token
+        "User-Agent": user
     })
     return res
 
@@ -54,7 +57,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, mode="text2img")
+    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: