Update app.py

app.py

@@ -10,4 +10,27 @@ examples = [
     ["Paris is the capital of France"]
 ]
 
-gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased", inputs=gr.inputs.Textbox(lines=5, label="Input Text"),title=title,description=description,article=article, examples=examples).launch(enable_queue=True)
+io1 = gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased")
+
+io2 = gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased-squad-qg")
+
+
+def inference(text, model):
+    if model == "prophetnet-large-uncased":
+        outtext = io1(text)
+    else:
+        outtext = io2(text)
+    return outtext   
+    
+gr.Interface(
+    inference, 
+    [gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["prophetnet-large-uncased","prophetnet-large-uncased-squad-qg"], type="value", default="prophetnet-large-uncased", label="model")
+], 
+    gr.outputs.Textbox(label="Output"),
+    examples=examples,
+    article=article,
+    title=title,
+    description=description).launch(enable_queue=True, cache_examples=True)
+
+
+