oza75 commited on
Commit
90c70f7
·
verified ·
1 Parent(s): 74a0c61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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