update

main.py

@@ -168,7 +168,6 @@ def waka(preface: str = "", author: str = "", first_line: str = "あききぬと
     Prompt: {waka_prompt}<br>
     Max New Tokens: {max_new_tokens}<br>
     """
-    yield info + "Generating Poem..."
     generated = __generate(wakagpt_tokenizer, wakagpt_model, waka_prompt, do_sample, num_beams, num_beam_groups, max_new_tokens, temperature, top_k, top_p, repetition_penalty, num_return_sequences)
 
     removed = 0
@@ -188,7 +187,7 @@ def waka(preface: str = "", author: str = "", first_line: str = "あききぬと
         checked_generated = [first_line + i[len(waka_prompt) - 1:] for i in generated]
 
     generated = [f"<p>{i}</p>" for i in checked_generated]
-    yield info + f"Removed Malformed: {removed}<br>Results:<br>{''.join(generated)}"
+    return info + f"Removed Malformed: {removed}<br>Results:<br>{''.join(generated)}"
 
 
 if __name__ == "__main__":