Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,7 @@ from transformers import BlenderbotForConditionalGeneration
|
|
| 6 |
|
| 7 |
@st.experimental_singleton
|
| 8 |
def get_models():
|
| 9 |
-
#
|
| 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)
|