Update fn.py

fn.py

@@ -26,6 +26,9 @@ def load_model():
     )
 
 def check(image):
+    if not image:
+        return None
+
     images = [image]
     safety_checker_input = feature_extractor(images, return_tensors="pt").to(device)
     images_np = [np.array(img) for img in images]