test

app.py

@@ -42,7 +42,7 @@ def infer(prompt, n_samples, steps, scale, 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'])
@@ -51,7 +51,7 @@ def infer(prompt, n_samples, steps, scale, seed):
             images.append(im)
 
     response = requests.post(url, json={
-        "model": "SafeStableDiffusion",
+        "model": "StableDiffusion",
         "prompt": prompt,
         "n": int(n_samples),
         "height": 512,