TNK21 commited on
Commit
ee4bb30
·
1 Parent(s): 6dc1020

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,9 +2,10 @@ import gradio as gr
2
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
3
 
4
  # Load the model and tokenizer
5
- model_name = "facebook/mbart-large-50-many-to-many-mmt"
6
- model = M2M100ForConditionalGeneration.from_pretrained(model_name)
7
- tokenizer = M2M100Tokenizer.from_pretrained(model_name)
 
8
 
9
  # Define the Gradio interface
10
  def translate_text(text, target_language):
 
2
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
3
 
4
  # Load the model and tokenizer
5
+ tokenizer = M2M100Tokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
6
+
7
+ # Initialize the model
8
+ model = M2M100ForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
9
 
10
  # Define the Gradio interface
11
  def translate_text(text, target_language):