Spaces:

tanyuzhou
/

Ro-Play

Sleeping

tanyuzhou commited on Jun 11

Commit

02bf74c

•

1 Parent(s): 4958e5f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def respond(
     # Decode the generated response
     response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
-    response = response.split("星野\n")[-1].strip()
     print("response: ", tokenizer.decode(generate_ids[0], skip_special_tokens=True))

     # Decode the generated response
     response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
+    response = response.split(f"老师\n{message}\n星野\n")[1]
     print("response: ", tokenizer.decode(generate_ids[0], skip_special_tokens=True))