Upload folder using huggingface_hub

app.py

@@ -7,7 +7,6 @@ from pathlib import Path
 from typing import TYPE_CHECKING, Iterable
 
 import gradio as gr
-import spaces
 from gradio.themes import Soft
 from gradio.themes.utils import colors, fonts, sizes
 from PIL import Image as PILImage
@@ -516,9 +515,8 @@ def L(key: str, lang: str = "en") -> str:
     return LABELS.get(lang, LABELS["en"]).get(key, key)
 
 
-@spaces.GPU
 def get_search_engine() -> MultimodalSearch:
-    """Get or create the search engine (lazy loading with GPU).
+    """Get or create the search engine (lazy loading).
 
     Returns:
         MultimodalSearch instance.
@@ -530,10 +528,10 @@ def get_search_engine() -> MultimodalSearch:
         from core.index import FaissIndex
         from core.search import MultimodalSearch
 
-        # Initialize embedding model with GPU
-        embedding_model = EmbeddingModel(device="cuda")
+        # Initialize embedding model (CPU for stability)
+        embedding_model = EmbeddingModel(device="cpu")
 
-        # Load FAISS index (CPU - faiss-cpu doesn't support GPU)
+        # Load FAISS index (CPU)
         index = FaissIndex(device="cpu")
         index_path = EMBEDDINGS_DIR / "image_index"
         index.load(index_path)
@@ -615,9 +613,8 @@ def format_results(results: list[dict]) -> list[tuple[PILImage.Image | str, str]
     return formatted
 
 
-@spaces.GPU
 def search_by_text_handler(query: str, top_k: int) -> list[tuple[str, str]]:
-    """Handle text search requests with GPU acceleration.
+    """Handle text search requests.
 
     Args:
         query: Text query string.
@@ -637,11 +634,10 @@ def search_by_text_handler(query: str, top_k: int) -> list[tuple[str, str]]:
         raise gr.Error(f"Search failed: {e}")
 
 
-@spaces.GPU
 def search_by_image_handler(
     image: PILImage.Image | None, top_k: int
 ) -> list[tuple[str, str]]:
-    """Handle image search requests with GPU acceleration.
+    """Handle image search requests.
 
     Args:
         image: Query image (PIL Image).
@@ -661,7 +657,6 @@ def search_by_image_handler(
         raise gr.Error(f"Search failed: {e}")
 
 
-@spaces.GPU
 def search_composed_handler(
     image: PILImage.Image | None,
     modification_text: str,
@@ -743,7 +738,6 @@ def get_random_samples(top_k: int = 10) -> list[tuple[PILImage.Image | str, str]
         return []
 
 
-@spaces.GPU
 def unified_search_handler(
     text_query: str,
     image: PILImage.Image | None,

requirements.txt

@@ -8,6 +8,3 @@ numpy>=1.24
 datasets>=2.14,<4.0
 huggingface-hub>=0.20
 tqdm>=4.65
-
-# HF Spaces ZeroGPU
-spaces