Update app.py

app.py

@@ -25,9 +25,9 @@ class UnifiedAssistant:
                 trust_remote_code=True
             )
 
-            # Initialize Docs Assistant (DocGPT)
+            # Initialize Docs Assistant (Using Mistral instead of DocGPT)
             print("Loading Documentation Assistant Model...")
-            self.docs_model_name = "Arc53/docsgpt-40b-falcon"
+            self.docs_model_name = "mistralai/Mistral-7B-Instruct-v0.2"
             self.docs_tokenizer = AutoTokenizer.from_pretrained(
                 self.docs_model_name,
                 trust_remote_code=True
@@ -68,7 +68,7 @@ class UnifiedAssistant:
             inputs = self.code_tokenizer(query, return_tensors="pt").to(self.code_model.device)
             outputs = self.code_model.generate(
                 **inputs,
-                max_length=2048,
+                max_length=20488,
                 temperature=0.7,
                 top_p=0.95,
                 do_sample=True