Merge pull request #27 from asapdiscovery/registry_updates

Enable registry updates
asapdiscovery · Sep 20, 2024 · b73bbff · b73bbff
2 parents d899c95 + 94eacd0
commit b73bbff
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/falcbot/falcbot.py b/falcbot/falcbot.py
@@ -60,6 +60,9 @@ def pred_matcher(event, logger, context):
 
 @app.event("app_mention", matchers=[pred_matcher])
 def make_pic50_pred(event, say, context, logger):
+    # refresh the registry every time a prediction is made
+    ASAPMLModelRegistry.update_registry()
+
     content = event.get("text")
     # parse with LLM
     worked, model = llm._BASIC_ML_LLM.query(content)