Update app.py

app.py

@@ -19,22 +19,14 @@ class ParrotletRetriever:
 
         # Load tokenizer and model
         self.tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
-        self.model = AutoModel.from_pretrained(model_name, trust_remote_code=True, token=HF_TOKEN)
+        self.model = AutoModel.from_pretrained(model_name, trust_remote_code=True, token=os.getenv("HF_TOKEN"))
         self.model.to(self.device)
         self.model.eval()
 
-        # Pinecone initialization (unchanged)
         self.pinecone_namespace = os.environ.get("NAMESPACE")
         self.pinecone_client = Pinecone(api_key=os.environ.get("PINECONE_API_KEY"))
         self.pinecone_index = self.pinecone_client.Index(host=os.environ.get("PINECONE_HOST"))
 
-        print(f"Connected to Pinecone index: {index_name}")
-        if self.pinecone_namespace:
-            print(f"🔹 Using namespace: {self.pinecone_namespace}")
-
-    # --------------------------
-    # Mean Pooling
-    # --------------------------
     def mean_pooling(self, model_output, attention_mask):
         """Mean pooling for sentence embeddings."""
         token_embeddings = model_output[0]