LamaAl commited on
Commit
88bb44d
1 Parent(s): 83eaa8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ from transformers import BlenderbotForConditionalGeneration
6
 
7
  @st.experimental_singleton
8
  def get_models():
9
- # it may be necessary for other frameworks to cache the model
10
- # seems pytorch keeps an internal state of the conversation
11
  model_name = "facebook/blenderbot-400M-distill"
12
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
13
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
6
 
7
  @st.experimental_singleton
8
  def get_models():
9
+ # pytorch keeps an internal state of the conversation
 
10
  model_name = "facebook/blenderbot-400M-distill"
11
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
12
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)