Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 import logging
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-import time
 
 # Set the logger to display only CRITICAL messages
 logging.basicConfig(level=logging.CRITICAL)
 
+
 # Cache the model and tokenizer to avoid reloading it every time
 @st.experimental_singleton
 def load_model():
@@ -32,18 +32,6 @@ if st.button("Submit"):
     if user_input:
         # Generate text based on the input
         generated_text = generate_text(user_input)
-        
-        # Initialize or reset the progress state
-        if 'text_progress' not in st.session_state:
-            st.session_state['text_progress'] = ""
-        
-        # Simulate typing effect
-        for i in range(len(generated_text)):
-            st.session_state['text_progress'] = generated_text[:i+1]
-            st.write(st.session_state['text_progress'])
-            time.sleep(0.05)  # Adjust as needed for the desired speed of the "typing" effect
-            
-        # Reset progress for next input
-        del st.session_state['text_progress']
+        st.write(generated_text)
     else:
-        st.write("Please enter a prompt.")
+        st.write("Please enter a prompt.")