After generating the images, clear the GPU cache

app.py

@@ -217,8 +217,10 @@ def generate(
         with open(f"{filename}.txt", "w") as f:
             f.write(prompt)
 
-    return output_images  # return the list of image objects
+    # After generating the images, clear the GPU cache
+    torch.cuda.empty_cache()
 
+    return output_images  # return the list of image objects