Spaces:
Runtime error
Runtime error
add logger
Browse files
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 |
|