Update app.py

app.py

@@ -32,9 +32,12 @@ def predict(prompt, history):
  )
  generated_text = sequences[0]['generated_text']
 
- final_result = generated_text[len(formatted_prompt):]
- return generated_text.strip()
- # return final_result.strip()
+ final_result = generated_text.split("### Assistant:")[1]
+ if " Human: " in final_result:
+ final_result = final_result.split(" Human: ")[0]
+ 
+ # return generated_text.strip()
+ return final_result.strip()
 
 gr.ChatInterface(predict,
  title="Tinyllama_chatBot",