Update app.py

app.py

@@ -18,8 +18,8 @@ class ParrotletRetriever:
         print(f"🚀 Loading model on {self.device}...")
 
         # 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=os.getenv("HF_TOKEN"))
+        self.tokenizer = AutoTokenizer.from_pretrained(model_name, token=os.getenv("HF_TOKEN"))
+        self.model = AutoModel.from_pretrained(model_name, token=os.getenv("HF_TOKEN"))
         self.model.to(self.device)
         self.model.eval()