Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,9 +21,10 @@ else:
|
|
21 |
logger.info("Using CPU for inference.")
|
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 |
-
|
|
|
27 |
translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
|
28 |
logger.info("Translation pipeline initialized successfully.")
|
29 |
|
|
|
21 |
logger.info("Using CPU for inference.")
|
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 |
+
revision=None
|
28 |
translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
|
29 |
logger.info("Translation pipeline initialized successfully.")
|
30 |
|