Update app.py

app.py

@@ -23,7 +23,7 @@ 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']).split('[s]')[1].split('[/s]')[0]
+  return_answer = (result[0]['generated_text']).replace(input_text,'').replace('[INST]','').replace('[/INST]','').replace('<s>','').replace('>','')
   return return_answer
     
 HTML_TEMPLATE = """