10:30

app.py

@@ -32,13 +32,15 @@ def generate_response(prompt):
     )
     token = outputs.sequences[0, input_length:]
     full_response = tokenizer.decode(token, skip_special_tokens=True)
-    
-    if "\n" in full_response:
-        response_lines = full_response.split("\n")
-        response = "\n".join(line for line in response_lines if "<bot>:" not in line)
+
+    if "<human>" in full_response:
+        trimmed_response = full_response.split("<human>")[0].strip()
     else:
-        response = full_response.strip()
-    return response
+        trimmed_response = full_response.strip()
+
+    logging.debug(f"Trimmed response: {trimmed_response}")
+    return trimmed_response
+
 
 @app.route("/", methods=["GET"])
 def home():