Spaces:

AgentSearch
/

AgentBase-Platform

Running

Arastun commited on 13 days ago

Commit

fc8a76d

1 Parent(s): 8189da6

fix: streamlit cloud pandas upcasting error

Files changed (1) hide show

pages/interface.py CHANGED Viewed

@@ -138,7 +138,7 @@ class AgentBaseUI:
         else:
             raise ValueError(f"Selected model must be one of {self.retrieval_models}")
-        self.agents_df["scores"] = 0 # reset
         agent_ids, _ = zip(*res)
         filtered_df = self.agents_df.loc[self.agents_df.agent_id.isin(agent_ids)]
         for index, row in filtered_df.iterrows():

         else:
             raise ValueError(f"Selected model must be one of {self.retrieval_models}")
+        self.agents_df["scores"] = 0.0
         agent_ids, _ = zip(*res)
         filtered_df = self.agents_df.loc[self.agents_df.agent_id.isin(agent_ids)]
         for index, row in filtered_df.iterrows():