Update app.py

app.py

@@ -25,9 +25,9 @@ class UnifiedAssistant:
                 trust_remote_code=True
             )
 
-            # Initialize Docs Assistant (Using Mistral instead of DocGPT)
+            # Initialize Docs Assistant (Using Zephyr instead)
             print("Loading Documentation Assistant Model...")
-            self.docs_model_name = "mistralai/Mistral-7B-Instruct-v0.2"
+            self.docs_model_name = "HuggingFaceH4/zephyr-7b-beta"
             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=20488,
+                max_length=2048,
                 temperature=0.7,
                 top_p=0.95,
                 do_sample=True
@@ -195,7 +195,7 @@ def create_app():
         ### Your AI-Powered Documentation & Code Assistant
         
         This application combines three powerful AI models:
-        - 💻 Cde Assistant (Qwen2.5-Coder-32B)
+        - 💻 Code Assistant (Qwen2.5-Coder-32B)
         - 📚 Documentation Helper (DocGPT-40B)
         - 📑 PDF Analyzer (Llama-3.3-70B)
         """)