Update app.py

app.py

@@ -8,6 +8,14 @@ models=[
     "nadiamaqbool81/llama-2-7b-int4-java-code-1.178k",
     "nadiamaqbool81/llama-2-7b-int4-python-code-510"
 ]
+names=[
+    "nadiamaqbool81/starcoderbase-java",   
+    "nadiamaqbool81/starcoderbase-python",
+    "nadiamaqbool81/codet5-java",
+    "nadiamaqbool81/codet5-python",    
+    "nadiamaqbool81/llama-2-java",
+    "nadiamaqbool81/llama-2-python"
+]
 model_box=[
     gr.load(f"models/{models[0]}"),
     gr.load(f"models/{models[1]}"),
@@ -24,7 +32,7 @@ def the_process(input_text, model_choice):
     return(output)
 
 gr.HTML("""<h1 style="font-weight:600;font-size:50;margin-top:4px;margin-bottom:4px;text-align:center;">Text to Code Generation</h1></div>""")
-model_choice = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", interactive=True)
+model_choice = gr.Dropdown(label="Select Model", choices=[m for m in names], type="index", interactive=True)
 input_text = gr.Textbox(label="Input Prompt")
 output_window = gr.Code(label="Generated Code")