Update app.py

app.py

@@ -103,10 +103,12 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
+from transformers import AutoTokenizer, AutoModel  
+
 class HuggingFaceEmbeddings(Embeddings):
     def __init__(self, model_name: str):
         self.tokenizer = AutoTokenizer.from_pretrained("FacebookAI/xlm-roberta-large")
-        self.model = AutoModelForMaskedLM.from_pretrained("FacebookAI/xlm-roberta-large")
+        self.model = AutoModel.from_pretrained("FacebookAI/xlm-roberta-large")  # Use AutoModel instead of AutoModelForMaskedLM
 
     def embed_documents(self, texts: List[str]) -> List[List[float]]:
         embeddings = []