Update src/streamlit_app.py

src/streamlit_app.py

@@ -122,7 +122,7 @@ clip_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14", cache_di
 clip_model.eval()
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14", use_fast=True, cache_dir="/tmp/hf_cache")
 
-text_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', cache_folder="/tmp/sbert")
+text_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', cache_folder="/tmp/sbert", device="cpu")
 
 model_id = "google/gemma-3-4b-it"
 model = Gemma3ForConditionalGeneration.from_pretrained(