Update app.py

app.py

@@ -72,13 +72,12 @@ def get_outputs(text, model_id_idx, task_id_idx):
     if pipe is None or full_id != loaded_model_id:
         loaded_model_id = full_id
         pipe = pipeline("summarization" if 'summarization' in full_id else "text2text-generation", full_id)
+    if 'summarization' in full_id:
+        out = pipe(text)[0]["summary_text"]
+        if 'wiki' in full_id:
+            out = out.replace("'''", " ")
     else:
-        if 'summarization' in full_id:
-            out = pipe(text)[0]["summary_text"]
-            if 'wiki' in full_id:
-                out = out.replace("'''", " ")
-        else:
-            out = pipe(text)[0]["generated_text"]
+        out = pipe(text)[0]["generated_text"]
     return out, model_link(full_id)