Update app.py

app.py

@@ -9,6 +9,8 @@ from rudolph.pipelines import generate_codebooks, self_reranking_by_image, self_
 from rudolph.pipelines import  zs_clf
 import gradio as gr
 from rudolph import utils
+from PIL import Image
+
 device = 'cpu'
 
 model = get_rudolph_model('350M',  fp16=True, device=device)