p

content_gen.py

@@ -51,7 +51,11 @@ def generate_content(prompt, max_new_tokens, num_beams, repetition_penalty, no_r
         repetition_penalty=repetition_penalty,
         no_repeat_ngram_size=no_repeat_ngram_size,
         do_sample=do_sample,
-        early_stopping=True
+        early_stopping=False
     )
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
+    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Trim to the last period
+    if '.' in output_text:
+        last_period = output_text.rfind('.')
+        output_text = output_text[:last_period+1]
+    return output_text