BounharAbdelaziz commited on
Commit
4e0cddf
·
verified ·
1 Parent(s): 041c6fb

added token

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -10,10 +10,12 @@ MODEL_PATHS = {
10
  "Terjman-Supreme-v2": "BounharAbdelaziz/Terjman-Supreme-v2.0"
11
  }
12
 
 
 
13
 
14
  # Translation function for Nano and Large models
15
  def translate_nano_large(text, model_path):
16
- translator = pipeline("translation", model=model_path)
17
  translated = translator(
18
  text,
19
  max_length=512,
@@ -29,8 +31,8 @@ def translate_nano_large(text, model_path):
29
 
30
  # Translation function for Ultra and Supreme models
31
  def translate_ultra_supreme(text, model_path):
32
- model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
33
- tokenizer = AutoTokenizer.from_pretrained(model_path, src_lang="eng_Latn", tgt_lang="ary_Arab")
34
  translator = pipeline(
35
  "translation",
36
  model=model,
 
10
  "Terjman-Supreme-v2": "BounharAbdelaziz/Terjman-Supreme-v2.0"
11
  }
12
 
13
+ # Load environment token
14
+ TOKEN = os.environ['TOKEN']
15
 
16
  # Translation function for Nano and Large models
17
  def translate_nano_large(text, model_path):
18
+ translator = pipeline("translation", model=model_path, token=TOKEN)
19
  translated = translator(
20
  text,
21
  max_length=512,
 
31
 
32
  # Translation function for Ultra and Supreme models
33
  def translate_ultra_supreme(text, model_path):
34
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_path, token=TOKEN)
35
+ tokenizer = AutoTokenizer.from_pretrained(model_path, src_lang="eng_Latn", tgt_lang="ary_Arab", token=TOKEN)
36
  translator = pipeline(
37
  "translation",
38
  model=model,