Spaces:

TrungNQ
/

qa_roberta

Runtime error

TrungNQ commited on Mar 24, 2024

Commit

a3ac2b0

verified ·

1 Parent(s): 6874f75

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,12 +126,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer,BlenderbotForCondit
 import torch
-#chat_tkn = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-#mdl = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
-chat_tkn = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 def converse(user_input, chat_history=[]):

 import torch
+chat_tkn = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+mdl = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+#chat_tkn = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+#mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 def converse(user_input, chat_history=[]):