Update app.py

app.py

@@ -1,10 +1,15 @@
+##################################### Imports ######################################
 import gradio as gr
 
+
+########################### Global objects and functions ###########################
 def greet(model_name, prompt_template, name):
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
 model_choices = ["gpt2", "bert-base-uncased", "llama3-8b"]
 
+
+##################################### App UI #######################################
 with gr.Blocks() as demo:
     gr.Markdown("# Instruction Tuning with Unsloth")
     model_name = gr.Dropdown(label="Model", choices=model_choices, value="gpt2")
@@ -18,4 +23,8 @@ with gr.Blocks() as demo:
                     inputs=[model_name, prompt_template, name_input], 
                     outputs=output)
 
-demo.launch()
+
+##################################### Launch #######################################
+
+if ___name__ == "__main__":
+    demo.launch()