feat: add log

src/streamlit_app.py

@@ -45,6 +45,7 @@ def main():
         #     user_prompt, tokenize=False, add_generation_prompt=True)
         # prompt = user_prompt
         # outputs = processor(prompt)
+        print('start')
         outputs = processor(user_prompt,
                             max_new_tokens=4096,
                             use_cache=True,
@@ -55,6 +56,7 @@ def main():
                             )
         response = outputs[0]["generated_text"][len(user_prompt):]
         st.write(response)
+        print('end')
 
 
 if __name__ == "__main__":