Update app.py

app.py

@@ -40,7 +40,7 @@ def load_memory(filename='chat_memory.pkl'):
 session_memory = load_memory()
 
 # ---- Response Generation ----
-def generate_response(prompt, max_length=35):
+def generate_response(prompt, max_length=25):
     inputs = tokenizer(prompt, return_tensors='pt', padding=True, truncation=True, max_length=max_length)
     input_ids = inputs['input_ids'].to(device)
     attention_mask = inputs['attention_mask'].to(device)
@@ -67,7 +67,7 @@ def generate_response(prompt, max_length=35):
     parts = response.split("\n", 1)
     if len(parts) > 1:
         before_indent = parts[0].strip()
-        after_indent = "vß Response:" + parts[1].strip()
+        after_indent = "vß Gertrude" + parts[1].strip()
         final_response = before_indent + '\n' + after_indent
     else:
         final_response = response.strip()
@@ -80,7 +80,7 @@ def advanced_agi_chat(user_input):
     save_memory(session_memory)
 
     # Generate the response based on the prompt
-    prompt = f"User: {user_input}\nAutistic-Gertrude:"
+    prompt = f"User: {user_input}\nResponse:"
     response = generate_response(prompt)
 
     return response