Update app.py

app.py

@@ -13,9 +13,7 @@ generator = pipeline("text2text-generation" model = "mrm8488/t5-small-finetuned-
 default_value = "How to put out a grease fire?"
 sent = st.text_area("Input", default_value, height = 275)
 num_return_sequences = st.sidebar.number_input('Number of Sequences to be Generated', min_value=1, max_value=5, value=1, step=1)
-def infer():
-  return generator(sent, num_return_sequences)
-outputs = infer(sent, num_return_sequences)
+outputs = generator(sent, num_return_sequences)
 st.write("Augmented Data:")
 for output in outputs[0]:
   st.write(outputs[0]["generated_text"])