Skip to content

Commit abd3f90

Browse files
authored
Replace debug prints with logger (#13)
1 parent 808a0c7 commit abd3f90

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

no-ocr-api/np_ocr/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def ai_search(user_query: str = Form(...), user_id: str = Form(...), case_name:
180180

181181
dataset = load_from_disk(dataset_path)
182182
search_results_data = []
183-
print(search_results)
183+
logger.info(search_results)
184184
for point in search_results:
185185
logger.info(point)
186186
score = point["_distance"]

no-ocr-api/np_ocr/search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def call_vllm(image_data: PIL.Image.Image, user_query: str, base_url: str, api_k
160160
}}
161161
"""
162162

163-
print(prompt)
163+
logger.info(prompt)
164164
buffered = BytesIO()
165165
max_size = (512, 512)
166166
image_data.thumbnail(max_size)

0 commit comments

Comments
 (0)