Update app.py

app.py

@@ -1,3 +1,32 @@
 import gradio as gr
 
-gr.load("models/nadiamaqbool81/starcoderbase-1b-hf_python").launch()
+models=[
+    "nadiamaqbool81/starcoderbase-1b-hf",   
+    "nadiamaqbool81/starcoderbase-1b-hf_python",
+    "nitrosocke/redshift-diffusion",    
+]
+model_box=[
+    gr.Interface.load(f"models/{models[0]}",live=True,preprocess=True),
+    gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
+    gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
+]
+current_model=model_box[0]
+
+def the_process(input_text, model_choice):
+    a_variable = model_box[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;">Example Space to copy/paste</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")
+
+    the_button.click(the_process, inputs=[input_text, model_choice], outputs=[output_window])
+
+demo.launch()