Update app.py

app.py

@@ -76,7 +76,7 @@ def embed_seq(model_ref, seq, chunk):
         with torch.no_grad():
             out = mdl(**{k: v.to(mdl.device) for k, v in toks.items()})
         vecs.append(out.last_hidden_state[:, 0, :].cpu().numpy())
-    return np.mean(vecs, axis=0, keepdims=True)
+    return np.mean(vecs, axis=0)
 
 @st.cache_resource
 def load_go_info():