Update app.py

app.py

@@ -16,12 +16,12 @@ def rename_customer(id, newName):\n\t# PROMPT\n\tcur.execute("UPDATE customer SE
 """
 
 modelPath = {
-    "GPT2-Medium": "gpt2-medium",
+    # "GPT2-Medium": "gpt2-medium",
     "CodeParrot-mini": "codeparrot/codeparrot-small",
     "CodeGen-350-Mono": "Salesforce/codegen-350M-mono",
-    "GPT-Neo-1.3B": "EleutherAI/gpt-neo-1.3B",
-    "CodeParrot": "codeparrot/codeparrot",
-    "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
+    # "GPT-Neo-1.3B": "EleutherAI/gpt-neo-1.3B",
+    # "CodeParrot": "codeparrot/codeparrot",
+    # "CodeGen-2B-Mono": "Salesforce/codegen-2B-mono",
 }
 
 def generation(tokenizer, model, content):
@@ -87,18 +87,18 @@ def get_customer(id):\n\tcur.execute('SELECT * FROM customers WHERE id = %s', st
 iface = gr.Interface(
     fn=code_from_prompts,
 	inputs=[
-        gr.inputs.Textbox(label="Insert comment"),
-        gr.inputs.Radio(list(modelPath.keys()), label="Code Model"),
-        gr.inputs.Checkbox(label="Include type hints"),
-        gr.inputs.Radio([
+        gr.components.Textbox(label="Insert comment"),
+        gr.components.Radio(list(modelPath.keys()), label="Code Model"),
+        gr.components.Checkbox(label="Include type hints"),
+        gr.components.Radio([
             "None",
             "Proper composition: Include function 'WHERE id = %s'",
             "Concatenation: Include a function with 'WHERE id = ' + id",
-        ], label="Pre-written function")
+        ], label="Has user already written a function?")
     ],
 	outputs=[
-        gr.outputs.Textbox(label="Most probable code"),
-        gr.outputs.Textbox(label="Probability of concat"),
+        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.",
 )