Spaces:
Sleeping
Sleeping
Valerioweb
commited on
Commit
·
2b5ef4f
1
Parent(s):
70ceed9
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
|
|
7 |
|
8 |
def launch(input):
|
9 |
tokenizer.src_lang = "en"
|
10 |
-
encoded_hi = tokenizer(
|
11 |
generated_tokens = model.generate(**encoded_hi, forced_bos_token_id=tokenizer.get_lang_id("it"))
|
12 |
response = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
13 |
return response
|
|
|
7 |
|
8 |
def launch(input):
|
9 |
tokenizer.src_lang = "en"
|
10 |
+
encoded_hi = tokenizer(input, return_tensors="pt")
|
11 |
generated_tokens = model.generate(**encoded_hi, forced_bos_token_id=tokenizer.get_lang_id("it"))
|
12 |
response = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
13 |
return response
|