Spaces:

Valerioweb
/

translate

Sleeping

Valerioweb commited on Sep 11, 2023

Commit

70ceed9

1 Parent(s): 4d836c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
-tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
-model = AutoModelForCausalLM.from_pretrained("facebook/m2m100_418M")
 def launch(input):
     tokenizer.src_lang = "en"

+from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 import gradio as gr
 import torch
+tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
+model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
 def launch(input):
     tokenizer.src_lang = "en"