Update app.py

app.py

@@ -21,9 +21,8 @@ model.load_adapter(peft_model_folder)
 def generate_dialogue(input_text):
 
   pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer,max_length=200)
-  result = pipe(f"<s>[INST] {input_text} [/INST]")
-
-  return_answer = (result[0]['generated_text']).replace(input_text,'').replace('[INST]','').replace('[/INST]','').replace('<s>','').replace('>','')
+  result = pipe(f"{input_text}")
+  return_answer = (result[0]['generated_text']).replace(input_text,'')
   return return_answer
     
 HTML_TEMPLATE = """