Increased temperature and added alternate openings

app.py

@@ -748,8 +748,8 @@ discrim_weights=None
 discrim_meta=None
 class_label=0
 length=100
-stepsize=0.99
-temperature=1.0
+stepsize=0.32
+temperature=1.3
 top_k=2
 sample=True
 num_iterations=10
@@ -811,17 +811,22 @@ tokenizer = GPT2Tokenizer.from_pretrained(pretrained_model)
 for param in model.parameters():
     param.requires_grad = False
 
+starters = ["How are you feeling and why?", "Tell me about your day", "What would you like to talk about?"]
 eot_token = "<|endoftext|>"
 
-def get_reply(response, username = None, histories = {}, in_stepsize = 0.99, in_horizon_length = 1, in_num_iterations = 10, in_top_k = 2):
+def get_reply(response, username = None, histories = {}, in_stepsize = 0.32, in_horizon_length = 1, in_num_iterations = 10, in_top_k = 2):
     if username == None or username == "": return "<div class='chatbot'>Enter a username</div>", histories
     stepsize = in_stepsize
     horizon_length = int(in_horizon_length)
     num_iterations = int(in_num_iterations)
     top_k = int(in_top_k)
-    if response.endswith(("bye", "Bye", "bye.", "Bye.", "bye!", "Bye!")):
-        histories[username] = None
-        return "<div class='chatbot'>Chatbot restarted</div>", histories
+    if response.endswith(("bye", "Bye", "bye.", "Bye.", "bye!", "Bye!","Hello", "Hi", "hello")):
+        starter = choice(starters)
+        histories[username] = starter+"<|endoftext|>"
+        html = "<div class='chatbot'> Chatbot restarted"
+        html += "<div class='msg user'>"+starter+"</div>"
+        html += "</div>"
+        return html, histories
     history = histories.get(username, None)
     convo_hist = (history if history != None else "How are you?<|endoftext|>") + response + eot_token
     # figure out conditioning text
@@ -877,8 +882,12 @@ def get_reply(response, username = None, histories = {}, in_stepsize = 0.99, in_
             convo_hist = eot_token.join(convo_hist_split)
 
         except:
-            histories[username] = None
-            return "<div class='chatbot'>Error occured, chatbot restarted</div>", histories
+            starter = choice(starters)
+            histories[username] = starter+"<|endoftext|>"
+            html = "<div class='chatbot'> Chatbot restarted"
+            html += "<div class='msg user'>"+starter+"</div>"
+            html += "</div>"
+            return html, histories
     histories[username] = convo_hist
     return html, histories
 
@@ -896,4 +905,4 @@ gr.Interface(fn=get_reply,
                     gr.inputs.Textbox(label="Username"),
                     "state"],
              outputs=["html", "state"],
-             css=css).launch(debug=True, enable_queue=True) 
+             css=css).launch(debug=True, enable_queue=True, share=True)