merve HF Staff commited on
Commit
d6ae831
Β·
1 Parent(s): 292c000

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,9 +1,9 @@
1
  from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
2
  import gradio as gr
3
 
4
- tokenizer = BlenderbotTokenizer.from_pretrained(model_id)
5
- model = BlenderbotForConditionalGeneration.from_pretrained(model_id)
6
- model_id = 'facebook/blenderbot-400M-distill'
7
 
8
 
9
  def infer(message):
 
1
  from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
2
  import gradio as gr
3
 
4
+ tokenizer = BlenderbotTokenizer.from_pretrained('facebook/blenderbot-400M-distill')
5
+ model = BlenderbotForConditionalGeneration.from_pretrained('facebook/blenderbot-400M-distill')
6
+
7
 
8
 
9
  def infer(message):