added logger for preprocessor

Dungeon-MASSters · Oct 14, 2023 · 85babc6 · 85babc6
1 parent 746d813
commit 85babc6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model/preprocessor/libs/models.py b/model/preprocessor/libs/models.py
@@ -14,6 +14,7 @@
 from transformers import BlipProcessor, BlipForConditionalGeneration, pipeline
 import torch
 import shutil
+from libs import logger
 
 class Translator:
     def __init__(self) -> None: