AhmadHazem commited on
Commit
1129545
·
1 Parent(s): 1a8dedb
Files changed (1) hide show
  1. app.py +2 -2
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(model_name_translate)
37
- model_translate = MarianMTModel.from_pretrained(model_path_translate, state_dict=tensors, config="lora-opus-mt-en-ar/config.json")
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):