Update app.py

app.py

@@ -13,10 +13,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 API_URL = os.environ.get("API_URL")
 
 with open("./HHH_prompt.txt", "r") as f:
-    HHH_PROMPT = f.read()
-
-#response = requests.get("https://huggingface.co/spaces/bigcode/bigcode-playground/resolve/main/HHH_prompt.txt")
-#HHH_PROMPT  = response.text
+    HHH_PROMPT = f.read() + "\n\n"
 
 FIM_PREFIX = "<fim_prefix>"
 FIM_MIDDLE = "<fim_middle>"
@@ -113,7 +110,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
     if fim_mode:
         output = prefix
     elif chat_mode:
-        output = prompt #chat_prompt
+        output = chat_prompt
     else:
         output = prompt