Spaces:

abrar-adnan
/

GeekBot

Sleeping

abrar-adnan commited on Nov 3, 2025

Commit

a2de0b4

verified ·

1 Parent(s): b29e29e

Update agents/research_agent.py

Files changed (1) hide show

agents/research_agent.py CHANGED Viewed

@@ -13,7 +13,7 @@ class ResearchAgent:
         print("Initializing RelevanceChecker with lightweight Hugging Face model...")
         # Use a smaller, CPU-friendly model by default
-        model_name = getattr(settings, "HF_MODEL_RELEVANCE", "google/flan-t5-small")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"

         print("Initializing RelevanceChecker with lightweight Hugging Face model...")
         # Use a smaller, CPU-friendly model by default
+        model_name = getattr(settings, "HF_MODEL_NAME")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"