error message

app.py

@@ -47,7 +47,8 @@ def infer(prompt, n_samples, steps, scale, seed):
     if 'output' not in data:
         raise gr.Error("Although safety guidance is enabled, potential unsafe content found. Please try again with different seed.")
     else:
-        if data['output']['result_type']
+        if data['output']['result_type'] == "error":
+            raise gr.Error(data['output']['value'])
         for image in data['output']['choices']:
             im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
             images.append(im)