AlexWortega commited on
Commit
b3f3de1
1 Parent(s): 7e551b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import torch
2
- from rudalle import get_tokenizer, get_vae
3
- from rudalle.utils import seed_everything
4
 
5
  import sys
6
 
@@ -20,7 +19,7 @@ import sys
20
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
21
  from tqdm import tqdm, trange
22
  import PIL.Image
23
- from IPython.display import Image
24
 
25
  import transformers
26
 
@@ -30,7 +29,7 @@ model_path = 'coco_prefix_latest.pt'
30
 
31
 
32
 
33
- #title Model
34
 
35
  class MLP(nn.Module):
36
 
 
1
  import torch
2
+
 
3
 
4
  import sys
5
 
 
19
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
20
  from tqdm import tqdm, trange
21
  import PIL.Image
22
+
23
 
24
  import transformers
25
 
 
29
 
30
 
31
 
32
+
33
 
34
  class MLP(nn.Module):
35