Increased temperature and added alternate openings

app.py

@@ -13,7 +13,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# print
+# temperature
 """
 Example command with bag of words:
 python examples/run_pplm.py -B space --cond_text "The president" --length 100 --gamma 1.5 --num_iterations 3 --num_samples 10 --stepsize 0.01 --window_length 5 --kl_scale 0.01 --gm_scale 0.95
@@ -28,7 +28,6 @@ import json
 from operator import add
 from typing import List, Optional, Tuple, Union
 from random import choice, randint
-from matplotlib import use
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -37,7 +36,6 @@ from tqdm import trange
 from transformers import GPT2Tokenizer
 from transformers.file_utils import cached_path
 from transformers.modeling_gpt2 import GPT2LMHeadModel
-
 from pplm_classification_head import ClassificationHead
 
 PPLM_BOW = 1
@@ -749,8 +747,8 @@ discrim_weights=None
 discrim_meta=None
 class_label=0
 length=100
-stepsize=5.12
-temperature=1.0
+stepsize=0.32
+temperature=1.3
 top_k=2
 sample=True
 num_iterations=0
@@ -765,8 +763,8 @@ seed=0
 no_cuda=False
 colorama=False
 verbosity="quiet"
-fp="./paper_code/discrim_models/persoothe_classifier.pt" #"/content/drive/Shareddrives/COS_IW04_ZL/COSIW04/Discriminators/3_class_lrggpt_fit_deeper_2/3_PerSoothe_classifier_head_epoch_8.pt"
-model_fp=None 
+fp="./paper_code/discrim_models/persoothe_classifier.pt"
+model_fp=None
 calc_perplexity=False
 is_deep=False
 is_deeper=True
@@ -812,17 +810,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 = 5.12, in_horizon_length = 1, in_num_iterations = 0, in_top_k = 2):
+def get_reply(response, username = None, histories = {}, in_stepsize = 0.32, in_horizon_length = 1, in_num_iterations = 0, 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
@@ -878,8 +881,12 @@ def get_reply(response, username = None, histories = {}, in_stepsize = 5.12, 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
 
@@ -893,8 +900,8 @@ css = """
 
 gr.Interface(fn=get_reply,
              theme="default",
-             inputs=[gr.inputs.Textbox(placeholder="How are you?"),
-                    gr.inputs.Textbox(label="Username"), 
+             inputs=[gr.inputs.Textbox(placeholder="How are you?"), 
+                    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)