Upload folder using huggingface_hub

backend/colpali.py

@@ -471,8 +471,9 @@ async def query_vespa_nearest_neighbor(
 
 def is_special_token(token: str) -> bool:
     # Pattern for tokens that start with '<', numbers, whitespace, or single characters, or the string 'Question'
+    # Will exclude these tokens from the similarity map generation
     pattern = re.compile(r"^<.*$|^\d+$|^\s+$|^\w$|^Question$")
-    if pattern.match(token):
+    if (len(token) < 3) or pattern.match(token):
         return True
     return False
 

frontend/app.py

@@ -279,7 +279,7 @@ def SearchResult(results: list, query_id: Optional[str] = None):
                         ),
                         Div(
                             Badge(
-                                "Relevance score: " + str(result["relevance"]),
+                                f"Relevance score: {result['relevance']:.4f}",
                                 cls="flex gap-1.5 items-center justify-center",
                             ),
                         ),