Update app.py

app.py

@@ -21,7 +21,8 @@ from torch.cuda.amp import autocast
 # Initialize the Flask app
 app = Flask(__name__)
 TEMP_DIR = None
-VIDEO_DIRECTORY = None
+VIDEO_DIRECTORY = os.path.abspath("videos")
+os.makedirs(VIDEO_DIRECTORY, exist_ok=True)
 
 def clear_cuda_memory():
     torch.cuda.empty_cache()
@@ -207,8 +208,8 @@ def generate_video():
         # Return the output video path or further process the file for download
         if final_output_video and final_output_video.endswith('.mp4'):
             filename = os.path.basename(final_output_video)
-            os.makedirs('videos', exist_ok=True)
-            VIDEO_DIRECTORY = os.path.abspath('videos')
+            # os.makedirs('videos', exist_ok=True)
+            # VIDEO_DIRECTORY = os.path.abspath('videos')
             print("VIDEO_DIRECTORY: ",VIDEO_DIRECTORY)
             destination_path = os.path.join(VIDEO_DIRECTORY, filename)
             shutil.copy(final_output_video, destination_path)