feat: add multiple model support

app.py

@@ -1,18 +1,25 @@
 import gradio as gr
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="pszemraj/distilgpt2-magicprompt-SD")
+models = {
+    "MagicPrompt": "pszemraj/distilgpt2-magicprompt-SD",
+    "Llama-SmolTalk-3.2-1B": "prithivMLmods/Llama-SmolTalk-3.2-1B-Instruct"
+}
+pipelines = {}
 
+for key, value in models.items():
+    pipelines[key] = pipeline("text-generation", model=value)
 
 def respond(
     message,
     _: list[tuple[str, str]],
+    model: str,
     max_new_tokens: int,
     temperature: float,
     top_p: float,
     top_k: int
 ):
-    yield pipe(
+    yield pipelines[model](
         message,
         max_new_tokens=max_new_tokens,
         do_sample=True,
@@ -28,6 +35,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
+        gr.Radio(choices=[models.items()], value="pszemraj/distilgpt2-magicprompt-SD", type="value", label="Model"),
         gr.Slider(minimum=8, maximum=128, value=64, step=8, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(