Spaces:

abrar-adnan
/

GeekBot

Sleeping

abrar-adnan commited on 19 days ago

Commit

48451c4

verified ·

1 Parent(s): f51eec3

Update agents/relevance_checker.py

Files changed (1) hide show

agents/relevance_checker.py CHANGED Viewed

@@ -11,7 +11,7 @@ class RelevanceChecker:
         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"