CodeParrot main model

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",
 }
 
@@ -29,6 +29,9 @@ preloadModels = {}
 for m in list(modelPath.keys()):
     preloadModels[m] = ecco.from_pretrained(modelPath[m])
 
+topComments = []
+rankings = requests.get("https://code-adv.herokuapp.com/db").json()['results']
+
 def generation(tokenizer, model, content):
     decoder = 'Standard'
     num_beams = 2 if decoder == 'Beam' else None
@@ -63,6 +66,7 @@ def generation(tokenizer, model, content):
     return [
         "".join(full_output.tokens),
         str(prob.item() * 100),
+        rankings
     ]
 
 def clean_comment(txt):
@@ -78,8 +82,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-small"])
-    model = preloadModels["CodeParrot-small"]
+    tokenizer = AutoTokenizer.from_pretrained(modelPath["CodeParrot"])
+    model = preloadModels["CodeParrot"]
 
     code = ""
     headerComment = headerComment.strip()
@@ -108,7 +112,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-small",
+            "model": "codeparrot/codeparrot",
             "headerComment": headerComment,
             "bodyComment": fnComment,
             "prefunction": pre_content,
@@ -134,7 +138,8 @@ iface = gr.Interface(
 	outputs=[
         gr.components.Textbox(label="Most probable code"),
         gr.components.Textbox(label="Probability of concat"),
+        gr.components.Json(value=rankings)
     ],
-	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",
+	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",
 )
 iface.launch()