Hugging Face's logo

Spaces:
muhammadayman
/
gradio-demo
Sleeping

App Files Community
muhammadayman commited on
Commit
0feee31
·
1 Parent(s): f725092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ import torch
11
 
12
  tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
13
  model = torch.load("helsinki_fineTuned.pt", map_location=torch.device('cpu'))
 
14
  #translation_pipeline = pipeline(model)
15
 
16
 
 
11
 
12
  tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
13
  model = torch.load("helsinki_fineTuned.pt", map_location=torch.device('cpu'))
14
+ model.eval()
15
  #translation_pipeline = pipeline(model)
16
 
17