Update app.py

app.py

@@ -1,6 +1,5 @@
 import torch
-from rudalle import get_tokenizer, get_vae
-from rudalle.utils import seed_everything
+
 
 import sys
 
@@ -20,7 +19,7 @@ import sys
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from tqdm import tqdm, trange
 import PIL.Image
-from IPython.display import Image 
+
 
 import transformers
 
@@ -30,7 +29,7 @@ model_path = 'coco_prefix_latest.pt'
 
 
 
-#title Model
+
 
 class MLP(nn.Module):