Update inference.py

inference.py

@@ -82,16 +82,16 @@ def generate_image_from_text_with_persistent_storage(caption, batch_size=4):
         # generate the images
         generated_images = generator(noise, sent_emb, eval=True).float()
 
-        # Create a temporary directory
-        with tempfile.TemporaryDirectory() as temp_dir:
-            pil_images = []
-            for idx, image_tensor in enumerate(generated_images.unbind(0)):
-                # Save the image tensor to a temporary file
-                image_path = os.path.join(temp_dir, f"image_{idx}.png")
-                torchvision.utils.save_image(image_tensor.data, image_path, value_range=(-1, 1), normalize=True)
-
-                # Load the saved image using Pillow
-                pil_image = Image.open(image_path)
-                pil_images.append(pil_image)
-
-            return pil_images
+        # Create a permanent directory if it doesn't exist
+        permanent_dir = "generated_images"
+        if not os.path.exists(permanent_dir):
+            os.makedirs(permanent_dir)
+
+        image_paths = []
+        for idx, image_tensor in enumerate(generated_images.unbind(0)):
+            # Save the image tensor to a permanent file
+            image_path = os.path.join(permanent_dir, f"image_{idx}.png")
+            torchvision.utils.save_image(image_tensor.data, image_path, value_range=(-1, 1), normalize=True)
+            image_paths.append(image_path)
+
+        return image_paths