Hugging Face's logo

Spaces:
smjain
/
chat
Runtime error

App Files Community
smjain commited on
Commit
9529237
·
1 Parent(s): 6854f2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import torch
7
 
8
  chat_tkn = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
9
- mdl = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
10
 
11
  def converse(user_input, chat_history=[]):
12
 
 
6
  import torch
7
 
8
  chat_tkn = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
9
+ mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
10
 
11
  def converse(user_input, chat_history=[]):
12