Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 
-generator = pipeline('text-generation', model='gpt2')
-
-def generate(text,max_length,temperature,num_beams,top_k,top_p,repetition_penalty):
+def generate(text,the_model,max_length,temperature,num_beams,top_k,top_p,repetition_penalty):
+    generator = pipeline('text-generation', model=the_model)
     result = generator(text, num_return_sequences=3,
     max_length=max_length,
     temperature=temperature,
@@ -18,6 +17,7 @@ demo = gr.Interface(
     fn=generate,
     inputs=[
         gr.Textbox(lines=5, label="Input Text"),
+        gr.Dropdown(choices=['gpt2','gpt2-medium','gpt2-large','gpt2-xl'],value = 'gpt2',label="Choose model"),
         gr.Slider(value=50,label="Max Length",minimum=1,maximum=200),
         gr.Slider(value=1.0,label="Temperature",minimum=0.0,maximum=1.0,step=0.05),
         gr.Slider(value=4,label="Num Beams",minimum=2,maximum=6,step=1),