Upload folder using huggingface_hub

model_trainer.py

@@ -18,6 +18,7 @@ def authenticate_hf(token: Optional[str]) -> None:
     else:
         print("Skipping Hugging Face login: HF_TOKEN not set.")
 
+@space.GPU
 def load_embedding_model(model_name: str) -> SentenceTransformer:
     """Initializes the Sentence Transformer model."""
     print(f"Loading Sentence Transformer model: {model_name}")

requirements.txt

@@ -7,3 +7,4 @@ gradio
 html_to_markdown
 sentence-transformers
 git+https://github.com/huggingface/transformers@v4.56.0-Embedding-Gemma-preview
+spaces