cut to User

app/api.py

@@ -132,6 +132,7 @@ async def text_generate(
     time_diff = time_end - time_start
     print(f"result:\n{result}")
     generated_text = result[len(prompt)+1:]
+    generated_text = generated_text[:generated_text.find("User:")]
     return {"generated_text": generated_text, "processing_time": time_diff}