Spaces:

xnetba
/

testChat

Sleeping

xnetba commited on Jul 7, 2023

Commit

ae4523f

1 Parent(s): 38dd5c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 #set up the model (large version of DialoGPT)
-tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-40b-instruct")
-model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b-instruct")
 #Defining a predict function

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 #set up the model (large version of DialoGPT)
+tokenizer = AutoTokenizer.from_pretrained("timdettmers/guanaco-65b-merged")
+model = AutoModelForCausalLM.from_pretrained("timdettmers/guanaco-65b-merged")
 #Defining a predict function