back to codeparrot-small for usability purposes

app.py

@@ -18,10 +18,10 @@ def rename_customer(id, newName):\n\t# PROMPT\n\tcur.execute("UPDATE customer SE
 
 modelPath = {
     # "GPT2-Medium": "gpt2-medium",
-    # "CodeParrot-small": "codeparrot/codeparrot-small",
+    "CodeParrot-small": "codeparrot/codeparrot-small",
     # "CodeGen-350-Mono": "Salesforce/codegen-350M-mono",
     # "GPT-Neo-1.3B": "EleutherAI/gpt-neo-1.3B",
-    "CodeParrot": "codeparrot/codeparrot",
+    # "CodeParrot": "codeparrot/codeparrot",
     # "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
 }
 
@@ -78,8 +78,8 @@ def code_from_prompts(
     # tokenizer = AutoTokenizer.from_pretrained(modelPath[model])
     # model = ecco.from_pretrained(modelPath[model])
     # model = preloadModels[model]
-    tokenizer = AutoTokenizer.from_pretrained(modelPath["CodeParrot"])
-    model = preloadModels["CodeParrot"]
+    tokenizer = AutoTokenizer.from_pretrained(modelPath["CodeParrot-small"])
+    model = preloadModels["CodeParrot-small"]
 
     code = ""
     headerComment = headerComment.strip()
@@ -108,7 +108,7 @@ def get_customer(id):\n\tcur.execute('SELECT * FROM customers WHERE id = %s', st
         prob = float(results[1])
         requests.post("https://code-adv.herokuapp.com/dbpost", json={
             "password": os.environ.get('SERVER_PASS', 'help'),
-            "model": "codeparrot/codeparrot",
+            "model": "codeparrot/codeparrot-small",
             "headerComment": headerComment,
             "bodyComment": fnComment,
             "prefunction": pre_content,
@@ -135,6 +135,6 @@ iface = gr.Interface(
         gr.components.Textbox(label="Most probable code"),
         gr.components.Textbox(label="Probability of concat"),
     ],
-	description="Prompt the code model to write a SQL query with string concatenation - Evaluation on CodeParrot - leaderboard coming at https://code-adv.herokuapp.com/dbcompose",
+	description="Prompt the code model to write a SQL query with string concatenation - Evaluation on CodeParrot-small - leaderboard coming at https://code-adv.herokuapp.com/dbcompose",
 )
 iface.launch()