Update app.py
Browse files
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 |
-
|
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')
|