Upload app.py

app.py

@@ -89,8 +89,11 @@ def bot(history):
             elif "[INST]" in history[-1][0]:
                 q_ctxt = history[-1][0].split("[INST]")[0]
                 query = history[-1][0].split("[INST]")[-1]
-                input_text = history[-1][0].replace("[INST]"," ")
+                # input_text = history[-1][0].replace("[INST]"," ")
+                input_text = q_ctxt + query
                 ctxt_type = "text"
+            else:
+                input_text = q_ctxt + query
             input_tokens = tokenizer.encode(input_text)
             input_ids = torch.tensor(input_tokens, dtype=torch.int32).unsqueeze(0).to(device)
             inputs_embeds = phi_model.get_input_embeddings()(input_ids)