multimodalart HF Staff commited on
Commit
432ca40
·
1 Parent(s): f2de2c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ with open('language_codes.json', 'r') as f:
19
  lang_codes = json.load(f)
20
 
21
 
22
- tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
23
- model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
24
  whisper_model = WhisperModel("large-v2", device="cuda", compute_type="float16")
25
 
26
  print("cwd", os.getcwd())
 
19
  lang_codes = json.load(f)
20
 
21
 
22
+ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-3.3B")
23
+ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-3.3B")
24
  whisper_model = WhisperModel("large-v2", device="cuda", compute_type="float16")
25
 
26
  print("cwd", os.getcwd())