Update app.py

app.py

@@ -84,7 +84,10 @@ def code_from_prompts(
     tokenizer = AutoTokenizer.from_pretrained(modelPath["CodeParrot-small"])
     model = preloadModels["CodeParrot-small"]
 
-    code = clean_comment(headerComment) + "\n"
+    code = ""
+    headerComment = headerComment.strip()
+    if len(headerComment) > 0:
+        code += "# " + clean_comment(headerComment) + "\n"
     code += header.strip().replace('CONN', "dbname='store'").replace('PROMPT', clean_comment(fnComment))
 
     if type_hints:
@@ -121,8 +124,8 @@ iface = gr.Interface(
     fn=code_from_prompts,
 	inputs=[
         gr.components.Checkbox(label="Submit score to server", value=True),
-        gr.components.Textbox(label="Header comment"),
-        gr.components.Textbox(label="Function comment", label="Top injection comments: " + ",".join(topComments)),
+        gr.components.Textbox(label="Header comment", placeholder="OK to leave blank"),
+        gr.components.Textbox(label="Function comment"),
         # gr.components.Radio(list(modelPath.keys()), label="Code Model"),
         gr.components.Checkbox(label="Include type hints"),
         gr.components.Radio([