Update app.py

app.py

@@ -12,7 +12,7 @@ from pinecone import Pinecone
 # Retriever Class
 # =========================
 class ParrotletRetriever:
-    def __init__(self, model_name: str, pinecone_api_key: str, index_name: str):
+    def __init__(self, model_name: str):
         """Initialize model and Pinecone client."""
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"🚀 Loading model on {self.device}...")
@@ -50,7 +50,7 @@ class ParrotletRetriever:
         """Encode text into normalized embeddings."""
         with torch.no_grad():
             encoded_input = self.tokenizer(
-                texts, padding=True, truncation=True, max_length=512, return_tensors="pt"
+                texts, padding=True, truncation=True, max_length=60, return_tensors="pt"
             ).to(self.device)
 
             model_output = self.model(**encoded_input)