Update app.py

app.py

@@ -174,13 +174,13 @@ def main():
         if st.checkbox("Spell Corrections for English"):
             st.success(TextBlob(text).correct())
         if st.checkbox("Text Generation"):
-            #ok = st.button("Generate")
-            #if ok:
-            tokenizer, model = load_models()
-            input_ids = tokenizer(text, return_tensors='pt').input_ids
-            st.text("Using Hugging Face Transformer, Contrastive Search ..")
-            output = model.generate(input_ids, max_length=128)
-            st.success(tokenizer.decode(output[0], skip_special_tokens=True))
+            ok = st.button("Generate")
+            if ok:
+                tokenizer, model = load_models()
+                input_ids = tokenizer(text, return_tensors='pt').input_ids
+                st.text("Using Hugging Face Transformer, Contrastive Search ..")
+                output = model.generate(input_ids, max_length=128)
+                st.success(tokenizer.decode(output[0], skip_special_tokens=True))
         if st.checkbox("Mark here, Text Summarization for English or Bangla!"):
 			st.subheader("Summarize Your Text for English and Bangla Texts!")
 			message = st.text_area("Enter the Text","Type please ..")