Update app.py

app.py

@@ -4,10 +4,10 @@ def get_pipe(name):
     tokenizer = AutoTokenizer.from_pretrained(name)
     model = AutoModelForSeq2SeqLM.from_pretrained(name)
     pipe = pipeline(
-	"zero-shot-classification", model=model, tokenizer=tokenizer, framework="pt"
+	"summarization", model=model, tokenizer=tokenizer, framework="pt"
     )
     return pipe
-model_names = ['facebook/bart-large-mnli'] #, 'bigscience/T0p', 'bigscience/T0pp']
+model_names = ['google/mt5-base'] #, 'bigscience/T0p', 'bigscience/T0pp']
 #model_names = ['bigscience/T0_3B','bigscience/T0'] #, 'bigscience/T0p', 'bigscience/T0pp']
 pipes = [get_pipe(name) for name in model_names]
 def _fn(text, do_sample, min_length, max_length, temperature, top_p, pipe):