Update api.py

api.py

@@ -199,10 +199,14 @@ def generate_text_greedy_strong(model, prompt, max_len=100, max_gen=98,
         next_token = np.argmax(next_logits)
         generated.append(int(next_token))  
 
-        decoded = sp.decode(generated)  
-        for t in ["<start>", "<sep>", "<end>"]:  
-            decoded = decoded.replace(t, "")  
-        decoded = decoded.strip()
+        decoded = sp.decode(generated)
+
+# <start>와 <sep> 제거
+        for t in ["<start>", "<sep>", "<end>"]:
+            decoded = decoded.replace(t, "")
+
+# 입력 프롬프트도 제거
+        decoded = decoded.replace(prompt, "").strip()
 
         if len(generated) >= min_len and (next_token == end_id or decoded.endswith(('요', '다', '.', '!', '?'))):  
             if is_greedy_response_acceptable(decoded):