Spaces:

xnetba
/

testChat

Sleeping

xnetba commited on Jul 7, 2023

Commit

3f04708

•

1 Parent(s): aa239df

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("nlpaueb/legal-bert-small-uncased")
-model = AutoModelForCausalLM.from_pretrained("nlpaueb/legal-bert-small-uncased")
 #Defining a predict function
 def predict(input, history=[]):

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