init

app.py

@@ -1,3 +1,17 @@
+# https://huggingface.co/spaces/kumshing/chinese-to-other
 import gradio as gr
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-gr.load("models/facebook/m2m100_418M").launch()
+
+def chineseToOther(name, lang = "en"):
+    tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
+    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
+    okenizer.src_lang = "zh"
+    encoded_zh = tokenizer(chinese_text, return_tensors="pt")
+    generated_tokens = model.generate(**encoded_zh, forced_bos_token_id=tokenizer.get_lang_id(lang))
+    translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    return translated_text
+
+demo = gr.Interface(fn=chineseToOther, inputs="text", outputs="text")
+demo.launch()