Update app.py

app.py

@@ -9,6 +9,8 @@ from torch import Tensor
 from transformers import AutoTokenizer, AutoModel
 from sklearn.metrics.pairwise import cosine_similarity
 
+import traceback
+
 
 def average_pool(last_hidden_states: Tensor,
                  attention_mask: Tensor) -> Tensor:
@@ -71,6 +73,7 @@ with gr.Blocks() as demo:
             print(f"retrieving #{message}")
             return _retrieve(message, chat_history)
         except:
+            print(traceback.format_exc())
             print(f"searching for {message}")
             return _search(message, chat_history)