Spaces:
Runtime error
Runtime error
Commit
·
e1e1d6a
1
Parent(s):
9b11bf4
Update app.py
Browse files
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-
|
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(
|
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)
|