Vageesh1 commited on
Commit
f56e8d2
1 Parent(s): d058d78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import torch
2
  import clip
3
  import PIL.Image
 
4
  import skimage.io as io
5
  import streamlit as st
6
  from transformers import GPT2Tokenizer, GPT2LMHeadModel, AdamW, get_linear_schedule_with_warmup
 
1
  import torch
2
  import clip
3
  import PIL.Image
4
+ from PIL import Image
5
  import skimage.io as io
6
  import streamlit as st
7
  from transformers import GPT2Tokenizer, GPT2LMHeadModel, AdamW, get_linear_schedule_with_warmup