Update app.py

app.py

@@ -85,7 +85,7 @@ def process_entity(batch, model, device):
 
 all_embeddings = []
 for batch in (data_loader):
-  embeddings, input_ids = process_entity(batch, model, device)
+  embeddings, input_ids = process_entity(batch, spaBERT_model, device)
   all_embeddings.append(embeddings)
 
 st.write("SpaBERT Embedding shape:", all_embeddings[0].shape)