Spaces:

xnetba
/

testChat

Sleeping

xnetba commited on Jul 7, 2023

Commit

3165746

1 Parent(s): 72aa0c4

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("facebook/blenderbot_small-90M")
-model = AutoModelForCausalLM.from_pretrained("facebook/blenderbot_small-90M")
 #Defining a predict function
 def predict(input, history=[]):

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