error message if output not available

app.py

@@ -44,9 +44,12 @@ def infer(prompt, n_samples, steps, scale, seed):
     response = 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))
-    for image in data['output']['choices']:
-        im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
-        images.append(im)
+    if 'output' not in data:
+        raise gr.Error("Although safety guidance is enabled, potential unsafe content found. Please try again with different seed.")
+    else:
+        for image in data['output']['choices']:
+            im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
+            images.append(im)
 
     # payload = {'prompt': prompt}
     # images_request = requests.post(url, json=payload)