Update model.py

model.py

@@ -9,8 +9,7 @@ from typing import Tuple, List, Union, Optional
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, AdamW, get_linear_schedule_with_warmup
 from tqdm import tqdm, trange
 import skimage.io as io
-import PIL.Image
-from IPython.display import Image 
+import PIL.Image 
 
 
 N = type(None)