Update app.py

app.py

@@ -4,8 +4,11 @@ import torch
 
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-tokenizer = GPT2Tokenizer.from_pretrained('dennis-fast/DialoGPT-ElonMusk')
-model = GPT2LMHeadModel.from_pretrained('dennis-fast/DialoGPT-ElonMusk')
+#model_name  = 'dennis-fast/DialoGPT-ElonMusk'
+model_name  = 'luca-martial/DialoGPT-Elon'
+
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model = GPT2LMHeadModel.from_pretrained(model_name)
 
 def predict(input, history=[]):
     # tokenize the new input sentence