Update app.py

app.py

@@ -6,6 +6,9 @@ import os, gc
 import gradio as gr
 from transformers import VitsModel, AutoTokenizer, pipeline
 import torch
+from transformers import pipeline
+
+text_cls = pipeline("text-classification", "ivanlau/language-detection-fine-tuned-on-xlm-roberta-base")
 
 model = VitsModel.from_pretrained("facebook/mms-tts-ind")
 tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ind")
@@ -26,7 +29,7 @@ def pred(product_name, orientation):
         orientation = "square"
         height = 1080
         width = 1080
-    folder_name, sentences = pexels.generate_videos(product_name, pexels_api_key, orientation, height, width, model, tokenizer)
+    folder_name, sentences = pexels.generate_videos(product_name, pexels_api_key, orientation, height, width, model, tokenizer, model2, tokenizer2, text_cls)
     gc.collect()
     utils.combine_videos(folder_name)