mikepastor11 commited on
Commit
a85e59b
·
verified ·
1 Parent(s): 176fa1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -33,7 +33,9 @@ SESSION_STARTED = False
33
  ##############################################################
34
  # Our model and tokenizer
35
  #
36
- MODEL_NAME = "facebook/blenderbot-400M-distill"
 
 
37
 
38
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
39
 
 
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