Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoMode
 
 from components import caption_chain, tag_chain
 from components import pexels, utils
-import os
+import os, gc
 import gradio as gr
 
 model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
@@ -26,6 +26,7 @@ pexels_api_key = os.getenv('pexels_api_key')
 
 def pred(product_name, orientation):
     folder_name, sentences = pexels.generate_videos(product_name, pexels_api_key, orientation, llm_chain, sum_llm_chain)
+    gc.collect()
     utils.combine_videos(folder_name)
     return ["\n".join(sentences), os.path.join(folder_name, "Final_Ad_Video.mp4")]
         #{'video':os.path.join(folder_name, "Final_Ad_Video.mp4"),