Spaces:

Valerioweb
/

translate

Sleeping

Valerioweb commited on Sep 11, 2023

Commit

4d836c0

1 Parent(s): 283305e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
 def launch(input):
     tokenizer.src_lang = "en"

 import torch
 tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
+model = AutoModelForCausalLM.from_pretrained("facebook/m2m100_418M")
 def launch(input):
     tokenizer.src_lang = "en"