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')
|