vsrinivas commited on
Commit
7d81dcd
·
verified ·
1 Parent(s): 5d40160

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -8,12 +8,11 @@ import gradio as gr
8
  logging.set_verbosity_error()
9
 
10
  asr = pipeline(task="automatic-speech-recognition",
11
- # model="./models/distil-whisper/distil-small.en")
12
  model="distil-whisper/distil-small.en")
13
 
14
  translator = pipeline(task="translation",
15
- # model="./models/facebook/nllb-200-distilled-600M",
16
- model="facebook/nllb-200-distilled-600M",
17
  torch_dtype=torch.bfloat16)
18
 
19
  flores_200_df = pd.read_csv("Flores200_language_codes.csv", encoding='cp1252')
 
8
  logging.set_verbosity_error()
9
 
10
  asr = pipeline(task="automatic-speech-recognition",
 
11
  model="distil-whisper/distil-small.en")
12
 
13
  translator = pipeline(task="translation",
14
+ model="facebook/nllb-200-3.3B",
15
+ # model="facebook/nllb-200-distilled-600M",
16
  torch_dtype=torch.bfloat16)
17
 
18
  flores_200_df = pd.read_csv("Flores200_language_codes.csv", encoding='cp1252')