Update app.py

app.py

@@ -1,7 +1,6 @@
 import datetime
 import streamlit as st
 import maritalk
-import time
 
 # 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?" 
 
@@ -12,20 +11,10 @@ def init_model():
     )
     return model
 
-# def chat(message, max_tokens=300):
-#     model = maritalk.MariTalk(
-#         key="100479706496023140829_534ece7d352abf00",
-#         model="sabia-3"  # No momento, suportamos os modelos sabia-3, sabia-2-medium e sabia-2-small
-#     )
-#     for response in model.generate(message, max_tokens=max_tokens)
-#     answer = response["answer"]
-#     return answer
-
 def stream_response(model, message, max_tokens=200):
     for response in model.generate(message, max_tokens=max_tokens, stream=True): 
         if 'text' in response.keys():
             yield response['text']
-        # time(0.05)
         
 # main()
 model = init_model()