Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 
 # Module imports
 from utilities.setup import get_json_cfg
+from utilities.templates import prompt_template
 
 ########################### Global objects and functions ###########################
 
@@ -25,7 +26,7 @@ def upload_visibility(radio):
 
 def greet(model_name, prompt_template, name, dataset):
     """The model call"""
-    return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
+    return f"Hello {name}!! Using model: {model_name} with template: {inject_prompt}"
 
 ##################################### App UI #######################################
 with gr.Blocks() as demo:
@@ -36,9 +37,13 @@ with gr.Blocks() as demo:
     ##### Model Inputs #####
 
     # Select Model
-    model_name = gr.Dropdown(label="Model", choices=conf['model']['choices'], value="gpt2")
+    modelnames = conf['model']['choices']
+    model_name = gr.Dropdown(label="Supported Models", 
+                             choices=modelnames, 
+                             value=modelnames[0])
     # Prompt template
-    prompt_template = gr.Textbox(label="Prompt Template", value="Instruction: {0}\nOutput: {1}")
+    inject_prompt = gr.Textbox(label="Prompt Template", 
+                                 value=prompt_template())
     # Prompt Input
     name_input = gr.Textbox(label="Your Name")
     # Dataset choice