Update app.py

app.py

@@ -50,10 +50,11 @@ def summ_bert(text):
 def para_t5(text):
     encoding = t5_para_tokenizer.encode(text, padding='longest', return_tensors='pt')
     outputs = t5_para_model.generate(
-                input_ids=encoding["input_ids"], attention_mask=encoding["attention_mask"],
-                max_length=512,
+                input_ids=encoding["input_ids"],
+                attention_mask=encoding["attention_mask"],
+                max_length=100,
                 do_sample=True,
-                top_k=200,
+                top_k=120,
                 top_p=0.95,
                 early_stopping=True,
                 num_return_sequences=5)
@@ -67,8 +68,8 @@ def para_t5(text):
 def summarize(text):
     t5_ = summ_t5(text)
     #bert_ = summ_bert(text)
-    para_ = para_t5(t5_)
-    return t5_, para_
+    #para_ = para_t5(t5_)
+    return t5_
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
@@ -79,6 +80,6 @@ if __name__ == "__main__":
             with gr.Column():
                 t5_output = gr.Textbox(label="T5 Base Output")
                 #bert_output = gr.Textbox(label="Bert Base Output")
-                para_output = gr.Textbox(label="T5 Paraphrase Output")
-        analyze_button.click(summarize, input_text, [t5_output, para_output])
+                #para_output = gr.Textbox(label="T5 Paraphrase Output")
+        analyze_button.click(summarize, input_text, [t5_output])
     demo.launch()