Update app.py

app.py

@@ -1,9 +1,6 @@
-import datetime
 import streamlit as st
 import maritalk
 
-# question = "Name the planets in the solar system? A: " # question = "Quais são os planetas do sistema solar?"  # question = "Qual é o maior planeta do sistema solar?" 
-
 def init_model():
     model = maritalk.MariTalk(
         key="100479706496023140829_534ece7d352abf00",
@@ -18,18 +15,10 @@ def stream_response(model, message, max_tokens=200):
         
 # main()
 model = init_model()
-before = datetime.datetime.now()
 with st.container():
     st.write('\n\n')
     st.title('LLM-LANAChat-Brasil\n\n')
     message = st.text_input("Chat", placeholder="E ai?")
     if message:
         max_tokens = 200
-        # for response in model.generate(message, max_tokens=max_tokens, stream=True): 
-        #     st.write_stream(response)
-        st.write_stream(stream_response(model, message, max_tokens))
-            
-after = datetime.datetime.now()
-current_time = (after - before) # .strftime("%H:%M:%S")
-print("\nTime Elapsed: ", current_time)
-st.write("\nTime Elapsed: ", current_time)   
+        st.write_stream(stream_response(model, message, max_tokens))