Spaces:

ysharma
/

Explore_llamav2_with_TGI

Running on CPU Upgrade

Narsil HF staff commited on Jul 18, 2023

Commit

06b82ab

•

1 Parent(s): f33dc3d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ css = """.toast-wrap { display: none !important } """
 def predict(message, chatbot):
-    input_prompt = f"[INST]<<SYS>>\n{system_message}\n<</SYS>>\n\n "
     for interaction in chatbot:
         input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "

 def predict(message, chatbot):
+    input_prompt = f"[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
     for interaction in chatbot:
         input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "