Update app.py

app.py

@@ -274,9 +274,10 @@ def generate(model, idx, max_new_tokens, context_size, tokenizer, text_to_token_
         # print(token_ids_to_text(idx_next, tokenizer))
         # Same as before: append sampled index to the running sequence
         idx = torch.cat((idx, idx_next), dim=1) # (batch_size, num_tokens+1)
-    new_idx = re.sub(".", ". Meow.", idx)
+    # new_idx = re.sub(".", ". Meow.", idx)
     
-    return new_idx
+    # return new_idx
+    return idx
 
 def text_to_token_ids(text, tokenizer):
     encoded = tokenizer.encode(text, allowed_special={'<|endoftext|>'})