Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 from transformers import pipeline
 import torch
 
-model = "xlnet/xlnet-base-cased"
+model = "GeneZC/MiniChat-2-3B"
 generator=pipeline(task='text-generation', model=model)
 
 
@@ -23,12 +23,11 @@ def generate(text, max_length):
 
 
 def respond(message, history, tone="natural", max_length=512):
-    prompt = f"Paraphrase this text in a more {tones[tone]} way: {message}"
+    prompt = f"<s> [|User|]Paraphrase this text in a more {tones[tone]} way: {message} </s>[|Assistant|]"
     text = generate(prompt, max_length)
     print(text)
     text = text[0]["generated_text"]
-    if prompt in text:
-        text = text.split(":", 1)[1]
+    text = text.split("[|Assistant|]", 1)[1]
     return text