Update app.py

app.py

@@ -387,8 +387,8 @@ model = None
 codes_emb = None
 
 def load_model(model_id):
-    global model
-    global codes_emb
+    # global model
+    # global codes_emb
     if model_id in model_ids[-2:]:
         model = CrossEncoder(
             # "jinaai/jina-reranker-v2-base-multilingual",
@@ -397,13 +397,15 @@ def load_model(model_id):
             automodel_args={"torch_dtype": "auto"},
             trust_remote_code=True,
         )
+        return model, _
     else:
         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=1)
         # codes_emb = np.mean([model.encode(x["examples"]) for x in examples], axis=1)
+        return model, codes_emb
         
-load_model(model_id)
+model, codes_emb = load_model(model_id)
 
 # for x in examples:
 #   codes_emb.append(model.encode(x["examples"]))
@@ -711,7 +713,7 @@ def reload(chosen_model_id):
     global codes_emb
 
     if chosen_model_id != model_id:
-        load_model(model_id)
+        model, codes_emb = load_model(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])