Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -22,8 +22,9 @@ else:
|
|
22 |
|
23 |
# Load the translation pipeline with the specified model and detected device
|
24 |
model_checkpoint = "oza75/bm-nllb-1.3B"
|
|
|
25 |
# model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
|
26 |
-
translator = pipeline("translation", model=model_checkpoint, device=device, max_length=512)
|
27 |
logger.info("Translation pipeline initialized successfully.")
|
28 |
|
29 |
# Define the languages supported
|
|
|
22 |
|
23 |
# Load the translation pipeline with the specified model and detected device
|
24 |
model_checkpoint = "oza75/bm-nllb-1.3B"
|
25 |
+
revision = "f55e4e7193eb95483b084248e5e4a91e28eb7dc2"
|
26 |
# model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
|
27 |
+
translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
|
28 |
logger.info("Translation pipeline initialized successfully.")
|
29 |
|
30 |
# Define the languages supported
|