kingabzpro commited on
Commit
f72e3da
1 Parent(s): 1b2f46a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ article = "<p style='text-align: center'><a href='https://medium.com/geekculture
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
  import torch
20
 
21
- tokenizer = AutoTokenizer.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
22
- model = AutoModelForCausalLM.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
23
 
24
  def predict(input, history=[]):
25
  # tokenize the new input sentence
 
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
  import torch
20
 
21
+ tokenizer = AutoTokenizer.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")
22
+ model = AutoModelForCausalLM.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")
23
 
24
  def predict(input, history=[]):
25
  # tokenize the new input sentence