Added second model

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import pipeline, set_seed
 
-chef = pipeline("text-generation", "nschenone/rap-distil")
+models = {
+    "Rap" : pipeline("text-generation", "nschenone/rap-distil"),
+    "Metal" : pipeline("text-generation", "nschenone/metal-distil")
+}
 
-def generate(text):
+def generate(text, model):
     max_length: int = 100
     num_beams: int = 5
     num_return_sequences: int = 1
@@ -14,7 +17,7 @@ def generate(text):
 
     set_seed(0)
 
-    generated = chef(
+    generated = models[model](
         text_inputs=text,
         max_length=max_length,
         num_beams=num_beams,
@@ -28,5 +31,12 @@ def generate(text):
     return generated[0]["generated_text"]
 
 
-iface = gr.Interface(fn=generate, inputs="text", outputs="text")
+iface = gr.Interface(
+    fn=generate,
+    inputs=[
+        "text",
+        gr.Dropdown(choices=models.keys(), value=models.keys()[0], label="Model")
+    ],
+    outputs="text"
+    )
 iface.launch()