Changed model used

app.py

@@ -754,7 +754,7 @@ top_k=2
 sample=True
 num_iterations=10
 grad_length=10000
-horizon_length=5
+horizon_length=1
 window_length=0
 decay=False
 gamma=1.0
@@ -765,7 +765,7 @@ 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="./paper_code/discrim_models/persoothe_encoder.pt" #None 
+model_fp=None 
 calc_perplexity=False
 is_deep=False
 is_deeper=True
@@ -813,7 +813,7 @@ for param in model.parameters():
 
 eot_token = "<|endoftext|>"
 
-def get_reply(response, history = None, in_stepsize = 2.56, in_horizon_length = 5, in_num_iterations = 10, in_top_k = 2):
+def get_reply(response, history = None, in_stepsize = 2.56, in_horizon_length = 1, in_num_iterations = 10, in_top_k = 2):
     stepsize = in_stepsize
     horizon_length = int(in_horizon_length)
     num_iterations = int(in_num_iterations)
@@ -891,7 +891,7 @@ gr.Interface(fn=get_reply,
              inputs=[gr.inputs.Textbox(placeholder="How are you?"), 
                     "state", 
                     gr.inputs.Number(default=2.56, label="Step"), 
-                    gr.inputs.Number(default=5, label="Horizon"), 
+                    gr.inputs.Number(default=1, label="Horizon"), 
                     gr.inputs.Number(default=10, label="Iterations"),
                     gr.inputs.Number(default=2, label="Top_k")],
              outputs=["html", "state"],

paper_code/discrim_models/persoothe_classifier.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cf08132d0dd06851a46aeb1174b1a3fdaa8f5cf1b80a369004c198a568859451
+oid sha256:dcf085c51358c345730c3ddea3ae4367a627d0aec83755ead6a50c199183f032
 size 3317769

paper_code/discrim_models/persoothe_encoder.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8a46c6d5025c717063b51d3508600ab2fffbcdadb74268821050feb4a15ca45e
-size 3134064907