Update app.py

app.py

@@ -259,9 +259,10 @@ def get_embedding_for_text(text, tokenizer, model):
             inputs = {k: v.to(model.device) for k, v in inputs.items()}
             
             with torch.no_grad():
-                outputs = model(**inputs)
+                # Get the correct output format
+                outputs = model(**inputs)[0]  # Access first element of tuple
             
-            embedding = outputs.last_hidden_state[:, 0, :].cpu().numpy()
+            embedding = outputs[:, 0, :].cpu().numpy()
             chunk_embeddings.append(embedding[0])
         except Exception as e:
             st.warning(f"Error processing chunk: {str(e)}")