[update]add code

main.py

@@ -71,7 +71,7 @@ def main():
         input_ids = torch.tensor([input_ids], dtype=torch.long)
         input_ids = input_ids.to(device)
 
-        output = ""
+        output: str = ""
         streamer = TextIteratorStreamer(tokenizer=tokenizer)
 
         generation_kwargs = dict(
@@ -95,7 +95,7 @@ def main():
             output_ = output_.replace("[UNK]", "")
             output_ = output_.replace(text, "")
 
-            output += output_
+            output += output_.strip()
             output_text_box.value += output
             yield output