Update app.py

app.py

@@ -71,14 +71,14 @@ def get_outputs(text, model_id_idx, task_id_idx):
     full_id = full_ids[task_id][model_id]
     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)
+    else:
         if 'summarization' in full_id:
-          pipe = pipeline("summarization", full_id)
-          out = pipe(text)[0]["summary_text"]
-          if 'wiki' in full_id:
-            out = out.replace("'''", " ")
+            out = pipe(text)[0]["summary_text"]
+            if 'wiki' in full_id:
+                out = out.replace("'''", " ")
         else:
-          pipe = pipeline("text2text-generation", full_id)
-          out = pipe(text)[0]["generated_text"]
+            out = pipe(text)[0]["generated_text"]
     return out, model_link(full_id)