Spaces:

kingabzpro
/

Rick_and_Morty_Bot

Sleeping

kingabzpro commited on Feb 19, 2022

Commit

f72e3da

1 Parent(s): 1b2f46a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,8 @@ article = "<p style='text-align: center'><a href='https://medium.com/geekculture
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-tokenizer = AutoTokenizer.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
-model = AutoModelForCausalLM.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
 def predict(input, history=[]):
     # tokenize the new input sentence

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