Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
import json
|
4 |
from transformers import pipeline
|
5 |
|
6 |
-
|
7 |
# Initialize the pipeline
|
8 |
try:
|
9 |
text_translator = pipeline("translation", model=model_path, torch_dtype=torch.bfloat16)
|
|
|
3 |
import json
|
4 |
from transformers import pipeline
|
5 |
|
6 |
+
model_path = pipeline("translation", model="facebook/nllb-200-distilled-600M",torch_dtype=torch.bfloat16)
|
7 |
# Initialize the pipeline
|
8 |
try:
|
9 |
text_translator = pipeline("translation", model=model_path, torch_dtype=torch.bfloat16)
|