reset timeout timer once a new token is generated

app.py

@@ -275,6 +275,8 @@ if user_input:
                 visible_response = re.sub(r"<think>.*?</think>", "", final_response, flags=re.DOTALL)
                 visible_response = re.sub(r"<think>.*$", "", visible_response, flags=re.DOTALL)
                 visible_placeholder.markdown(visible_response)
+                # Reset the timer after each token is generated
+                start_time = time.time()
             except queue.Empty:
                 if time.time() - start_time > timeout:
                     st.error("Response generation timed out.")