Update app.py

app.py

@@ -273,7 +273,7 @@ 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