Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-#model_name  = 'dennis-fast/DialoGPT-ElonMusk'
-model_name  = 'luca-martial/DialoGPT-Elon'
+model_name  = 'dennis-fast/DialoGPT-ElonMusk'
+#model_name  = 'luca-martial/DialoGPT-Elon'
 
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
@@ -19,13 +19,14 @@ def predict(input, history=[]):
     
     # generate a response 
     history = model.generate(bot_input_ids,
-                            max_length=200,
+                            max_length=1000,
+                            #max_length=200,
                             pad_token_id=tokenizer.eos_token_id,
-                            no_repeat_ngram_size=3,
-                            do_sample=True,
-                            top_k=100,
-                            top_p=0.7,
-                            temperature = 0.8
+                            #no_repeat_ngram_size=3,
+                            #do_sample=True,
+                            #top_k=100,
+                            #top_p=0.7,
+                            #temperature = 0.8
                             ).tolist()
 
     # convert the tokens to text, and then split the responses into the right format