update

app.py

@@ -27,7 +27,7 @@ def predict(name, sl, topk, topp):
             repetition_penalty=sl
         )[0]["generated_text"]
 
-        return x1[len(f"<|endoftext|>"):]+"\n\n"+x2[len(f"<|endoftext|>{new_name}\n\n"):].replace("[TEXT]", new_name)
+        return x1[len(f"<|endoftext|>"):]+"\n\n"+x2[len(f"<|endoftext|>{new_name}\n\n"):].replace(" [TEXT] ", new_name)
     else:
         x1 = tdk1(f"<|endoftext|>{name}\n\n",
             do_sample=True, 
@@ -46,7 +46,7 @@ def predict(name, sl, topk, topp):
             repetition_penalty=sl
         )[0]["generated_text"]
 
-        return x1[len(f"<|endoftext|>{name}\n\n"):]+"\n\n"+x2[len(f"<|endoftext|>{name}\n\n"):].replace(["[TEXT]", name])
+        return x1[len(f"<|endoftext|>{name}\n\n"):]+"\n\n"+x2[len(f"<|endoftext|>{name}\n\n"):].replace([" [TEXT] ", name])