Added exception handling on line 66

app.py

@@ -67,15 +67,18 @@ def transform_ncuda(img,prompt,cfg=8.0,stps=30,sc=0.8):
       print('----------------------------------------------------------------------------------')
       print(f'{resp}')
       print(f'DEBUG: Type = {resp.__class__}')
-      for artifact in resp.artifacts:
-          if artifact.finish_reason == generation.FILTER:
-              warnings.warn(
-                  "Your request activated the API's safety filters and could not be processed."
-                  "Please modify the prompt and try again.")
-          if artifact.type == generation.ARTIFACT_IMAGE:
-              global img2
-              img2 = Image.open(io.BytesIO(artifact.binary))
-              return img2
+      try:
+          for artifact in resp.artifacts:
+              if artifact.finish_reason == generation.FILTER:
+                  warnings.warn(
+                      "Your request activated the API's safety filters and could not be processed."
+                      "Please modify the prompt and try again.")
+              if artifact.type == generation.ARTIFACT_IMAGE:
+                  global img2
+                  img2 = Image.open(io.BytesIO(artifact.binary))
+                  return img2
+      except Exception as e:
+          print(f'Caught error: {e}')
             # img2.save(str(artifact.seed)+ "-img2img.png") # Save our generated image with its seed number as the filename and the img2img suffix so that we know this is our transformed image.