Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import os
 import re
 from tqdm import tqdm
+from typing import List
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast
 app = Flask(__name__)