Update app.py

app.py

@@ -6,9 +6,9 @@ models=[
     "nadiamaqbool81/starcoderbase-1b-hf_python",    
 ]
 model_box=[
-    gr.Interface.load(f"models/{models[0]}",preprocess=True),
-    gr.Interface.load(f"models/{models[1]}",preprocess=True),
-    gr.Interface.load(f"models/{models[2]}",preprocess=True),
+    gr.load(f"models/{models[0]}"),
+    gr.load(f"models/{models[1]}"),
+    gr.load(f"models/{models[2]}"),
 ]
 current_model=model_box[0]
 
@@ -17,16 +17,10 @@ def the_process(input_text, model_choice):
     output = a_variable(input_text)
     return(output)
 
-with gr.Blocks() as demo:
-    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>""")
-    with gr.Row():
-        with gr.Column():
-            model_choice = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", interactive=True)
-            input_text = gr.Textbox(label="Input Prompt")
-            the_button = gr.Button()
-        with gr.Column():
-            output_window = gr.Code(label="Generated Code")
+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")
 
-    the_button.click(the_process, inputs=[input_text, model_choice], outputs=[output_window])
-
-demo.launch()
+interface = gr.Interface(fn=the_process, inputs=[input_text, model_choice], outputs="text")
+interface.launch()