Spaces:
Running
on
Zero
Running
on
Zero
added token
Browse files
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,
|