Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
|
|
| 13 |
model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B").to(device)
|
| 14 |
model.eval()
|
| 15 |
|
| 16 |
-
|
| 17 |
class Language:
|
| 18 |
def __init__(self, name, code):
|
| 19 |
self.name = name
|
|
@@ -141,21 +141,9 @@ def trans_page(input,trg):
|
|
| 141 |
translated_text=input
|
| 142 |
pass
|
| 143 |
|
|
|
|
| 144 |
|
| 145 |
-
|
| 146 |
-
new_lang = []
|
| 147 |
-
tokenizer.src_lang = src_lang
|
| 148 |
-
for langs in lang_id:
|
| 149 |
-
with torch.no_grad():
|
| 150 |
-
encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
|
| 151 |
-
generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
|
| 152 |
-
trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=False)[0]
|
| 153 |
-
new_lang.append(trans_langs)
|
| 154 |
-
else:
|
| 155 |
-
new_lang=lang_id
|
| 156 |
-
pass
|
| 157 |
-
|
| 158 |
-
return translated_text,gr.Dropdown.update(choices=[l.name for l in new_lang]),gr.Dropdown.update(choices=[l.name for l in new_lang])
|
| 159 |
|
| 160 |
def trans_to(input,src,trg):
|
| 161 |
for lang in lang_id:
|
|
@@ -208,7 +196,7 @@ with gr.Blocks() as transbot:
|
|
| 208 |
message = gr.Textbox(label="Prompt",placeholder="Enter Prompt",lines=4)
|
| 209 |
translated = gr.Textbox(label="Translated",lines=4,interactive=False)
|
| 210 |
gr.Column()
|
| 211 |
-
t_submit.click(trans_page,[md,t_space],[md
|
| 212 |
|
| 213 |
submit.click(trans_to, inputs=[message,lang_from,lang_to], outputs=[translated])
|
| 214 |
transbot.queue(concurrency_count=20)
|
|
|
|
| 13 |
model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B").to(device)
|
| 14 |
model.eval()
|
| 15 |
|
| 16 |
+
|
| 17 |
class Language:
|
| 18 |
def __init__(self, name, code):
|
| 19 |
self.name = name
|
|
|
|
| 141 |
translated_text=input
|
| 142 |
pass
|
| 143 |
|
| 144 |
+
|
| 145 |
|
| 146 |
+
return translated_text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 147 |
|
| 148 |
def trans_to(input,src,trg):
|
| 149 |
for lang in lang_id:
|
|
|
|
| 196 |
message = gr.Textbox(label="Prompt",placeholder="Enter Prompt",lines=4)
|
| 197 |
translated = gr.Textbox(label="Translated",lines=4,interactive=False)
|
| 198 |
gr.Column()
|
| 199 |
+
t_submit.click(trans_page,[md,t_space],[md])
|
| 200 |
|
| 201 |
submit.click(trans_to, inputs=[message,lang_from,lang_to], outputs=[translated])
|
| 202 |
transbot.queue(concurrency_count=20)
|