Update app.py

app.py

@@ -22,7 +22,7 @@ class MemoryTracker:
     def clear_memory():
         gc.collect()
         torch.cuda.empty_cache() if torch.cuda.is_available() else None
-        
+
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(levelname)s - %(message)s'
@@ -257,7 +257,10 @@ def generate_response(user_message: str, conversation_history: List[Dict[str, st
     prompt_parts.append(f"User: {user_message}\nAssistant:")
     prompt = "\n\n".join(prompt_parts)
     
-    return llm_handler.generate_response(prompt)
+    # Increase max_length to accommodate longer inputs
+    max_length = 512  # You can adjust this value as needed
+    
+    return llm_handler.generate_response(prompt, max_length)
 
 def chatbot_interface(user_message: str, history: Optional[List[Dict[str, str]]] = None):
     if history is None:
@@ -365,4 +368,4 @@ with gr.Blocks(css=custom_css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)