Spaces:

Yuchan5386
/

InteractGPT-API

Sleeping

Yuchan5386 commited on 18 days ago

Commit

9d91b93

verified ·

1 Parent(s): 757cfac

Update api.py

Files changed (1) hide show

api.py CHANGED Viewed

@@ -201,8 +201,7 @@ def generate_text_typical(model, prompt, max_len=100, max_gen=98,
         decoded = decoded.strip()
         if len(generated) >= min_len and (sampled == end_id or decoded.endswith(('.', '!', '?'))):
-            yield decoded
-            break
 def is_valid_response(response):
     if len(response.strip()) < 2:

         decoded = decoded.strip()
         if len(generated) >= min_len and (sampled == end_id or decoded.endswith(('.', '!', '?'))):
+            return decoded  # ← yield 대신 return
 def is_valid_response(response):
     if len(response.strip()) < 2: