Update app.py

app.py

@@ -228,7 +228,7 @@ model_id = model_ids[-1]
 model = SentenceTransformer(model_id, trust_remote_code=True)
 
 # codes_emb = model.encode([x[6:] for x in codes])
-codes_emb = model.encode([x["examples"] for x in examples]).mean(axis=0)
+codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=0)
 
 
 def censor_middle(number, num_to_hide=4):
@@ -349,7 +349,7 @@ def respond(
     plate_numbers = ", ".join(plates)
     
     text_emb = model.encode(message)
-    scores = cos_sim(codes_emb, text_emb)[:,0]
+    scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
 
     if is_multiple:
         request_details = []
@@ -501,7 +501,7 @@ def reload(chosen_model_id):
         model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
         model_id = chosen_model_id
         # codes_emb = model.encode([x[6:] for x in codes])
-        codes_emb = model.encode([x["examples"] for x in examples]).mean(axis=0)
+        codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=0)
         return f"Model {chosen_model_id} has been succesfully loaded!"
     return f"Model {chosen_model_id} is ready!"