Update app.py

app.py

@@ -2,6 +2,7 @@ import tiktoken
 import torch
 import time
 import math
+import re
 from torch.utils.data import Dataset, DataLoader
 
 import gradio as gr