.

BlinkDL · Jul 1, 2024 · 6b78005 · 6b78005
1 parent a27ceb3
commit 6b78005
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/API_DEMO_CHAT.py b/API_DEMO_CHAT.py
@@ -87,13 +87,11 @@ def run_rnn(ctx):
 
     return out
 
-
-init_ctx = "User: hi" + "\n\n"
-init_ctx += "Assistant: Hi. I am your assistant and I will provide expert full response in full details. Please feel free to ask any question and I will always answer it." + "\n\n"
-
-run_rnn(init_ctx)
-
-print(init_ctx, end="")
+if STATE_NAME == None: # use initial prompt if we are not loading a state
+    init_ctx = "User: hi" + "\n\n"
+    init_ctx += "Assistant: Hi. I am your assistant and I will provide expert full response in full details. Please feel free to ask any question and I will always answer it." + "\n\n"
+    run_rnn(init_ctx)
+    print(init_ctx, end="")
 
 while True:
     msg = prompt("User: ")