Spaces:
Sleeping
Sleeping
Commit
•
9a9e016
1
Parent(s):
a85e59b
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,9 +33,9 @@ SESSION_STARTED = False
|
|
| 33 |
##############################################################
|
| 34 |
# Our model and tokenizer
|
| 35 |
#
|
| 36 |
-
|
| 37 |
|
| 38 |
-
MODEL_NAME = "facebook/blenderbot-3B"
|
| 39 |
|
| 40 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
| 41 |
|
|
|
|
| 33 |
##############################################################
|
| 34 |
# Our model and tokenizer
|
| 35 |
#
|
| 36 |
+
MODEL_NAME = "facebook/blenderbot-400M-distill"
|
| 37 |
|
| 38 |
+
# MODEL_NAME = "facebook/blenderbot-3B"
|
| 39 |
|
| 40 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
| 41 |
|