alymbeks commited on
Commit
fe45a00
·
1 Parent(s): 33971ff

add logger

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -4,6 +4,13 @@ import pickle
4
  import os
5
  import gradio as gr
6
  import zipfile
 
 
 
 
 
 
 
7
 
8
  # Load CLIP model
9
  text_model = SentenceTransformer("clip-ViT-B-32-multilingual-v1")
@@ -49,6 +56,7 @@ def search_text(query, top_k=1):
49
  Returns:
50
  [list]: [list of images that are related to the query.]
51
  """
 
52
  # First, we encode the query.
53
  query_emb = text_model.encode([query])
54
 
 
4
  import os
5
  import gradio as gr
6
  import zipfile
7
+ import logging
8
+
9
+ logger = logging.getLogger("query_search")
10
+ logger.setLevel(logging.INFO)
11
+ handler = logging.FileHandler('queries.log', 'a', 'utf-8')
12
+ handler.setFormatter(logging.Formatter('%(asctime)s query: %(message)s'))
13
+ logger.addHandler(handler)
14
 
15
  # Load CLIP model
16
  text_model = SentenceTransformer("clip-ViT-B-32-multilingual-v1")
 
56
  Returns:
57
  [list]: [list of images that are related to the query.]
58
  """
59
+ logger.info(f"{query}, {top_k}")
60
  # First, we encode the query.
61
  query_emb = text_model.encode([query])
62