nbroad HF staff commited on
Commit
f0ab314
1 Parent(s): ecae52c

add requirements

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +5 -0
app.py CHANGED
@@ -18,6 +18,7 @@ model.to(device)
18
 
19
 
20
  def detect(query):
 
21
  tokens = tokenizer.encode(query)
22
  all_tokens = len(tokens)
23
  tokens = tokens[:tokenizer.max_len - 2]
 
18
 
19
 
20
  def detect(query):
21
+ # Copied from https://github.com/openai/gpt-2-output-dataset/tree/master/detector#L35-L46
22
  tokens = tokenizer.encode(query)
23
  all_tokens = len(tokens)
24
  tokens = tokens[:tokenizer.max_len - 2]
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ transformers==2.0.0
2
+ tokenizers==0.7.0
3
+ requests
4
+ torch==1.4.0
5
+ gradio