Update app.py

app.py

@@ -3,12 +3,14 @@ import gradio as gr
 models=[
     "nadiamaqbool81/starcoderbase-1b-hf",   
     "nadiamaqbool81/starcoderbase-1b-hf_python",
-    "nadiamaqbool81/starcoderbase-1b-hf_python",    
+    "nadiamaqbool81/codet5-large-hf",
+    "nadiamaqbool81/codet5-large-hf-python",    
 ]
 model_box=[
     gr.load(f"models/{models[0]}"),
     gr.load(f"models/{models[1]}"),
     gr.load(f"models/{models[2]}"),
+    gr.load(f"models/{models[3]}"),
 ]
 current_model=model_box[0]
 
@@ -20,7 +22,7 @@ def the_process(input_text, model_choice):
 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)
 input_text = gr.Textbox(label="Input Prompt")
-output_window = gr.Textbox(label="Generated Code")
+output_window = gr.Code(label="Generated Code")
 
 interface = gr.Interface(fn=the_process, inputs=[input_text, model_choice], outputs="text")
 interface.launch()