Mgolo commited on
Commit
ce7b853
·
verified ·
1 Parent(s): 9bea29b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
21
 
22
  # --- Load translation models ---
23
  def load_models():
24
- en_dar_model_path = "/LocaleNLP/English_Hausa"
25
- en_wol_model_path = "/LocaleNLP/eng_wolof"
26
- en_hau_model_path = "/LocaleNLP/English_Darija"
27
 
28
  en_dar_model = AutoModelForSeq2SeqLM.from_pretrained(en_dar_model_path).to(device)
29
  en_dar_tokenizer = MarianTokenizer.from_pretrained(en_dar_model_path)
 
21
 
22
  # --- Load translation models ---
23
  def load_models():
24
+ en_dar_model_path = "LocaleNLP/english_hausa"
25
+ en_wol_model_path = "LocaleNLP/eng_wolof"
26
+ en_hau_model_path = "LocaleNLP/english_darija"
27
 
28
  en_dar_model = AutoModelForSeq2SeqLM.from_pretrained(en_dar_model_path).to(device)
29
  en_dar_tokenizer = MarianTokenizer.from_pretrained(en_dar_model_path)