Spaces:

dt
/

chatbot-es

Runtime error

Dabs commited on Feb 19, 2022

Commit

d0c5d85

1 Parent(s): 0a271b1

spanish

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 def predict(input, history=[]):
     # tokenize the new input sentence

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+tokenizer = AutoTokenizer.from_pretrained("ncoop57/DiGPTame-medium")
+model = AutoModelForCausalLM.from_pretrained("ncoop57/DiGPTame-medium")
 def predict(input, history=[]):
     # tokenize the new input sentence