feat: add max new tokens, validate title

app.py

@@ -23,7 +23,7 @@ def generate_story():
             gpt_out = kancilgpt.generate(
                 **gpt_input,
                 do_sample=True,
-                max_new_tokens=1,
+                max_new_tokens=16,
                 pad_token_id=gpt_tokenizer.eos_token_id,
                 eos_token_id=gpt_tokenizer.eos_token_id
             )
@@ -34,12 +34,17 @@ def generate_story():
             if len(splitted_result) <= 2:
                 _, judul_prompt = splitted_result
                 _, *judul_words = judul_prompt.split()
+
                 judul = " ".join(judul_words)
+                yield judul + "..."
+                if "." in judul:
+                    print("Invalid judul!")
+                    prompt = "<s> awal cerita | judul:"
+                    continue
+
                 isi = ""
                 end_part = ""
 
-                yield judul + "..."
-
                 if gpt_out[-1] == gpt_tokenizer.eos_token_id:
                     continue
             else:
@@ -64,7 +69,7 @@ def generate_story():
 
                     if quote_count % 2 != 0:
                         print("Invalid isi!")
-                        trimmed_isi = isi[:prev_i].strip()
+                        trimmed_isi = isi[:prev_i].rstrip()
                         prompt = f"<s> awal cerita | judul: {judul} | {trimmed_isi}"
                         continue