diff --git a/modules/shared.py b/modules/shared.py
index dcce9299..8ea3b441 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -5,6 +5,7 @@ import os
 import sys
 import time
 
+from PIL import Image
 import gradio as gr
 import tqdm