last touches

app.py

@@ -18,8 +18,11 @@ ARTICLE = r"""<center>
               For more please visit: <a href='https://sites.google.com/view/dr-gabriel-lopez/home'>My Page</a><br>
               </center>"""
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-model = TFAutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+# checkpoint = "ericzhou/DialoGPT-Medium-Rick_v2" #pytorch
+# checkpoint = "epeicher/DialoGPT-medium-homer" #pytorch
+checkpoint = "microsoft/DialoGPT-medium" #tf
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = TFAutoModelForCausalLM.from_pretrained(checkpoint)
 
 # interaction function
 def chat_with_bot(user_input, chat_history_and_input=[]):