Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from huggingface_hub import hf_hub_download
 import fasttext
 
 # Initialize fastText model
-model_path = 'lid.323.ftz'
-language_model = fasttext.load_model(model_path)
+# model_path = 'lid.323.ftz'
+# language_model = fasttext.load_model(model_path)
 
+lid_model = fasttext.load_model(hf_hub_download("facebook/fasttext-language-identification", "model.bin"))
 model_path_translation = "anzorq/m2m100_418M_ft_ru-kbd_44K"
 tgt_lang="zu"
 
@@ -14,7 +16,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_path_translation, use_safete
 
 def translate(text, num_beams=4, num_return_sequences=4):
     # Detect language
-    languages, _ = language_model.predict(text, k=1)
+    languages, _ = lid_model.predict(text, k=1)
     detected_language = languages[0].replace("__label__", "")
     
     inputs = tokenizer(text, return_tensors="pt")