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 |
|