Upload 2 files

app.py

@@ -93,7 +93,8 @@ class Prime:
         output = output[0].upper() + output[1:]
         if output[-1] == tokenizer.decode(stop_tokens[0]):
             output = output[:-1]
-        output = re.sub(" \d\.", "\n•", output)
+        if len(re.findall("\d\.", output)) > 1:
+            output = re.sub("\d\.", "<br>•", output)
         return output
 
 model.eval()
@@ -102,7 +103,7 @@ prime = Prime(tokenizer = tokenizer, model = model)
 
 def process_input(user_input, max_tokens, sample, top_k, message):
     return prime.generate(prompt = user_input, message = message, 
-                          max_tokens = 150, sample = sample,
+                          max_tokens = max_tokens, sample = sample,
                           top_k = top_k)
 
 
@@ -139,7 +140,7 @@ with gr.Blocks(title="primo", css="footer {visibility: hidden}", theme=gr.themes
             gr.Markdown("<b>options</b>")
             max_tokens = gr.Slider(1, 250, value=150, label="max tokens", info="choose a limit between 1 and 250")
             sample = gr.Checkbox(label="sampling")
-            top_k = gr.Slider(1, 5, step=1, value=1, label="creativity", info="choose a level between 1 and 5")
+            top_k = gr.Slider(1, 5, value=1, label="creativity", info="choose a level between 1 and 5")
             message = gr.Textbox(label="system message", value = "")
             clear = gr.Button("clear chat")
         with gr.Column(scale=8):

app_it.py

@@ -93,7 +93,9 @@ class Prime:
         output = output[0].upper() + output[1:]
         if output[-1] == tokenizer.decode(stop_tokens[0]):
             output = output[:-1]
-        output = re.sub(" \d\.", "\n•", output)
+        if len(re.findall("\d\.", output)) > 1:
+            output = re.sub("\d\.", "<br>•", output)
+        output = re.sub("^\<br\>", "", output)
         return output
 
 model.eval()
@@ -102,7 +104,7 @@ prime = Prime(tokenizer = tokenizer, model = model)
 
 def process_input(user_input, max_tokens, sample, top_k, message):
     return prime.generate(prompt = user_input, message = message, 
-                          max_tokens = 150, sample = sample,
+                          max_tokens = max_tokens, sample = sample,
                           top_k = top_k)