Spaces:

thenHung
/

demo_chatbot

Runtime error

thenHung commited on Jul 5, 2023

Commit

734d410

1 Parent(s): 30368ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,11 +10,10 @@ examples = [["How are you?"]]
 # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 # Load model directly
-tokenizer = AutoTokenizer.from_pretrained("rinna/vicuna-13b-delta-finetuned-langchain-MRKL")
-model = AutoModelForCausalLM.from_pretrained("rinna/vicuna-13b-delta-finetuned-langchain-MRKL")
-# tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-# model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 def user(message, history):

 # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 # Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 def user(message, history):