Spaces:

xnetba
/

testChat

Sleeping

xnetba commited on Jul 7, 2023

Commit

d9a1782

1 Parent(s): 969c331

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 #set up the model
-tokenizer = AutoTokenizer.from_pretrained("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
-model = AutoModelForCausalLM.from_pretrained("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
 #Defining a predict function
 def predict(input, history=[]):

 import torch
 #set up the model
+tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
+model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
 #Defining a predict function
 def predict(input, history=[]):