thenHung commited on
Commit
4ce38af
·
1 Parent(s): 9548c93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ examples = [["How are you?"]]
10
  # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
11
 
12
  # Load model directly
13
- tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
14
- model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
15
 
16
  def generate_response(input_text, chat_history=[]):
17
  # Tokenize the new input sentence
 
10
  # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
11
 
12
  # Load model directly
13
+ tokenizer = AutoTokenizer.from_pretrained("rinna/vicuna-13b-delta-finetuned-langchain-MRKL")
14
+ model = AutoModelForCausalLM.from_pretrained("rinna/vicuna-13b-delta-finetuned-langchain-MRKL")
15
 
16
  def generate_response(input_text, chat_history=[]):
17
  # Tokenize the new input sentence