fix: gradio version

README.md

@@ -4,7 +4,7 @@ emoji: 💡
 colorFrom: yellow
 colorTo: gray
 sdk: gradio
-sdk_version: 3.23.0
+sdk_version: 4.4.1
 app_file: app.py
 pinned: false
 license: mit

requirements.txt

@@ -4,11 +4,9 @@ pandas
 transformers
 sentence-transformers
 
+torch
+
 openpyxl
 tabulate
-gradio
 loguru
-
-torch
-
-
+# gradio

utils/models.py

@@ -20,6 +20,7 @@ list_models = [
 
 class SBert:
     def __init__(self, path):
+        logger.info(f'Start loading {self.__class__} from {path} ...')
         self.model = SentenceTransformer(path, device=DEVICE)
         logger.info(f'Load {self.__class__} from {path} ...')
 
@@ -31,6 +32,7 @@ class SBert:
 
 class ModelWithPooling:
     def __init__(self, path):
+        logger.info(f'Start loading {self.__class__} from {path} ...')
         self.tokenizer = AutoTokenizer.from_pretrained(path)
         self.model = AutoModel.from_pretrained(path)
         logger.info(f'Load {self.__class__} from {path} ...')