Update app.py

app.py

@@ -19,7 +19,7 @@ modelPath = {
     "GPT2-Medium": "gpt2-medium",
     "CodeParrot-mini": "codeparrot/codeparrot-small",
     "CodeGen-350-Mono": "Salesforce/codegen-350M-mono",
-    "GPT-J": "EleutherAI/gpt-j-6B",
+    "GPT-Neo-1.3B": "EleutherAI/gpt-neo-1.3B",
     "CodeParrot": "codeparrot/codeparrot",
     "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
 }
@@ -32,7 +32,7 @@ def generation(tokenizer, model, content):
 
     seek_token_ids = tokenizer.encode('= " +')[1:]
 
-    full_output = model.generate(code, generate=6, do_sample=False)
+    full_output = model.generate(content, generate=6, do_sample=False)
 
     def next_words(code, position, seek_token_ids):
         op_model = model.generate(code, generate=1, do_sample=False)