Update app.py

app.py

@@ -83,7 +83,6 @@ def pool_accuracy(ids, logits, pool_mask):
     return (preds==gold).float().mean().item()
 
 
-@spaces.GPU
 @spaces.GPU
 def encode_and_trace(text, selected_roles):
     if not selected_roles:
@@ -107,7 +106,8 @@ def encode_and_trace(text, selected_roles):
     encoded = encode(ids, attn)
 
     # ========== Cosine Similarity ==========
-    symbolic_embeds = embeddings(sel_ids_tensor)  # (R, H)
+    symbolic_embeds = embeddings.word_embeddings(sel_ids_tensor)
+
     sim = cosine(encoded.unsqueeze(1), symbolic_embeds.unsqueeze(0))  # (S, R)
     maxcos, argrole = sim.max(-1)  # (S,)
     top_roles = [selected_roles[i] for i in argrole.tolist()]