Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -23,8 +23,10 @@ else:
|
|
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-
|
27 |
-
revision=None
|
|
|
|
|
28 |
translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
|
29 |
logger.info("Translation pipeline initialized successfully.")
|
30 |
|
|
|
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/bm-nllb-600-01"
|
27 |
+
revision = None
|
28 |
+
# model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
|
29 |
+
# revision=None
|
30 |
translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
|
31 |
logger.info("Translation pipeline initialized successfully.")
|
32 |
|