Update app.py

app.py

@@ -78,19 +78,6 @@ except Exception as e:
 # Configure system message
 system_message = """You are a helpful AI assistant called AQuaBot. You provide direct, clear answers to questions while being aware of environmental impact. Keep your responses natural and concise. Respond directly to questions without using any special tags or markers."""
 
-def generate_response(user_input, chat_history):
-    try:
-        logger.info("Generating response for user input...")
-        global total_water_consumption
-
-        # Calculate water consumption for input
-        input_water_consumption = calculate_water_consumption(user_input, True)
-        total_water_consumption += input_water_consumption
-
-        # Create prompt following Llama 2 chat format
-        conversation = f""<s>[INST] {system_message}
-
-
 # Constants for water consumption calculation
 WATER_PER_TOKEN = {
     "input_training": 0.0000309,
@@ -135,7 +122,7 @@ def generate_response(user_input, chat_history):
             for message in chat_history:
                 conversation_history += f"[INST] {message[0]} [/INST] {message[1]} "
         
-        prompt = f"{system_message}\n\n{conversation_history}[INST] {user_input} [/INST]"
+        prompt = f"<s>[INST] {system_message}\n\n{conversation_history}[INST] {user_input} [/INST]"
 
         logger.info("Generating model response...")
         outputs = model_gen(
@@ -186,7 +173,7 @@ try:
             <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
                 <h1 style="color: #2d333a;">AQuaBot</h1>
                 <p style="color: #4a5568;">
-                    Welcome to AQuaBot - An AI assistant powered by Llama 2 that helps raise awareness 
+                    Welcome to AQuaBot - An AI assistant that helps raise awareness 
                     about water consumption in language models.
                 </p>
             </div>