AhmadHazem
commited on
Commit
·
1129545
1
Parent(s):
1a8dedb
fix
Browse files
app.py
CHANGED
@@ -33,8 +33,8 @@ with safe_open("model.safetensors", framework="pt") as f:
|
|
33 |
for k in f.keys():
|
34 |
tensors[k] = f.get_tensor(k)
|
35 |
|
36 |
-
tokenizer_translation = MarianTokenizer.from_pretrained(
|
37 |
-
model_translate = MarianMTModel.from_pretrained(
|
38 |
|
39 |
@spaces.GPU
|
40 |
def translate(sentence):
|
|
|
33 |
for k in f.keys():
|
34 |
tensors[k] = f.get_tensor(k)
|
35 |
|
36 |
+
tokenizer_translation = MarianTokenizer.from_pretrained(model_path_translate)
|
37 |
+
model_translate = MarianMTModel.from_pretrained(model_name_translate, state_dict=tensors, config="lora-opus-mt-en-ar/config.json")
|
38 |
|
39 |
@spaces.GPU
|
40 |
def translate(sentence):
|