mrsteyk commited on
Commit
75f32d9
1 Parent(s): 3f17815

I swear to god how did this even happen...

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,10 +2,13 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import gradio as gr
3
 
4
  prologue = """quality: high
 
5
  [System]
6
  Assistant is a distilled language model trained by the community.<|STK_SP|>
 
7
  [System]
8
  <|STK_SP|>
 
9
  [User]"""
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("mrsteyk/openchatgpt-neo-125m", use_fast=True)
 
2
  import gradio as gr
3
 
4
  prologue = """quality: high
5
+
6
  [System]
7
  Assistant is a distilled language model trained by the community.<|STK_SP|>
8
+
9
  [System]
10
  <|STK_SP|>
11
+
12
  [User]"""
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("mrsteyk/openchatgpt-neo-125m", use_fast=True)