Update app.py

app.py

@@ -14,17 +14,22 @@ io1 = gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased")
 
 io2 = gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased-squad-qg")
 
+io3 = gr.Interface.load("huggingface/microsoft/prophetnet-large-uncased-cnndm")
+
+
 
 def inference(text, model):
     if model == "prophetnet-large-uncased":
         outtext = io1(text)
-    else:
+    elif model == "prophetnet-large-uncased-squad-qg":
         outtext = io2(text)
+    else:
+        outtext = io3(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.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["prophetnet-large-uncased","prophetnet-large-uncased-squad-qg","prophetnet-large-uncased-cnndm"], type="value", default="prophetnet-large-uncased", label="model")
 ], 
     gr.outputs.Textbox(label="Output"),
     examples=examples,