Update app.py

app.py

@@ -66,9 +66,6 @@ def calculate_eou(chat_ctx, session):
     eou_token_id = tokenizer.encode("<|im_end|>")[-1]
     return probs[eou_token_id]
 
-# Read system message from file
-with open("character/herta.txt", "r") as f:
-    system_message = f.read()
 
 # Respond function
 def respond(
@@ -79,7 +76,7 @@ def respond(
     top_p,
 ):
     # Keep the last 4 conversation pairs (user-assistant)
-    messages = [{"role": "system", "content": system_message}]
+    messages = [{"role": "system", "content": os.environ.get('CHARACTER_DESC')}]
 
     for val in history[-10:]:  # Only use the last 4 pairs
         if val[0]: