Spaces:
Sleeping
Sleeping
Commit
·
06d85b4
1
Parent(s):
c47d401
Update app.py
Browse files
app.py
CHANGED
@@ -2,9 +2,13 @@ import sys
|
|
2 |
import gradio as gr
|
3 |
from transformers import AutoTokenizer
|
4 |
import torch
|
|
|
|
|
5 |
tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
|
6 |
model = torch.load("helsinki_fineTuned.pt", map_location=torch.device('cpu'))
|
7 |
model.eval()
|
|
|
|
|
8 |
def translate_gradio(input):
|
9 |
tokenized_text = tokenizer.prepare_seq2seq_batch([input], return_tensors='pt')
|
10 |
encode = model.generate(**tokenized_text)
|
|
|
2 |
import gradio as gr
|
3 |
from transformers import AutoTokenizer
|
4 |
import torch
|
5 |
+
|
6 |
+
|
7 |
tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
|
8 |
model = torch.load("helsinki_fineTuned.pt", map_location=torch.device('cpu'))
|
9 |
model.eval()
|
10 |
+
|
11 |
+
|
12 |
def translate_gradio(input):
|
13 |
tokenized_text = tokenizer.prepare_seq2seq_batch([input], return_tensors='pt')
|
14 |
encode = model.generate(**tokenized_text)
|