test3

Running on T4

Cagliostro commited on Mar 18

Commit

add553d

•

1 Parent(s): c2cfe45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,8 +170,8 @@ def generate(
         if images and IS_COLAB:
             for image in images:
-            filepath = utils.save_image(image, metadata, OUTPUT_DIR)
-            logger.info(f"Image saved as {filepath} with metadata")
         return images, metadata
     except Exception as e:

         if images and IS_COLAB:
             for image in images:
+                filepath = utils.save_image(image, metadata, OUTPUT_DIR)
+                logger.info(f"Image saved as {filepath} with metadata")
         return images, metadata
     except Exception as e: