Update app.py

app.py

@@ -66,21 +66,22 @@ def transform_ncuda(img,prompt,cfg=8.0,stps=30,sc=0.8):
 
   # Set up our warning to print to the console if the adult content classifier is tripped.
   # If adult content classifier is not tripped, display generated image.
-  for resp in answers2:
-      print('----------------------------------------------------------------------------------')
-      print(f'{resp}')
-      print(f'DEBUG: Type = {resp.__class__}')
-      try:
+  try:
+      for resp in answers2:
+          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.")
+                          "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:
+                      global img2
+                      img2 = Image.open(io.BytesIO(artifact.binary))
+                      return img2
+  except Exception as e:
           raise gr.Error("Input image from sketch")
           # print(f'Caught error: {e}')
           # logging.warn(f'Caught error: {e}')