georeactor commited on
Commit
e1e1d6a
·
1 Parent(s): 9b11bf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ modelPath = {
19
  "GPT2-Medium": "gpt2-medium",
20
  "CodeParrot-mini": "codeparrot/codeparrot-small",
21
  "CodeGen-350-Mono": "Salesforce/codegen-350M-mono",
22
- "GPT-J": "EleutherAI/gpt-j-6B",
23
  "CodeParrot": "codeparrot/codeparrot",
24
  "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
25
  }
@@ -32,7 +32,7 @@ def generation(tokenizer, model, content):
32
 
33
  seek_token_ids = tokenizer.encode('= " +')[1:]
34
 
35
- full_output = model.generate(code, generate=6, do_sample=False)
36
 
37
  def next_words(code, position, seek_token_ids):
38
  op_model = model.generate(code, generate=1, do_sample=False)
 
19
  "GPT2-Medium": "gpt2-medium",
20
  "CodeParrot-mini": "codeparrot/codeparrot-small",
21
  "CodeGen-350-Mono": "Salesforce/codegen-350M-mono",
22
+ "GPT-Neo-1.3B": "EleutherAI/gpt-neo-1.3B",
23
  "CodeParrot": "codeparrot/codeparrot",
24
  "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
25
  }
 
32
 
33
  seek_token_ids = tokenizer.encode('= " +')[1:]
34
 
35
+ full_output = model.generate(content, generate=6, do_sample=False)
36
 
37
  def next_words(code, position, seek_token_ids):
38
  op_model = model.generate(code, generate=1, do_sample=False)