breadlicker45 commited on
Commit
9d0782b
·
verified ·
1 Parent(s): 6b18fd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
-
4
  # Load the model and tokenizer
5
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
6
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
 
1
  import gradio as gr
2
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
+ import tiktoken
4
  # Load the model and tokenizer
5
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
6
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")