From 603df3c63dcbf6e6aca9abf3de54bd5df65c6f82 Mon Sep 17 00:00:00 2001
From: Nathan Cooper <nacooper01@email.wm.edu>
Date: Wed, 29 Sep 2021 21:38:14 -0400
Subject: [PATCH] Change instance check to image instead of imagefile

---
 sentence_transformers/models/CLIPModel.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sentence_transformers/models/CLIPModel.py b/sentence_transformers/models/CLIPModel.py
index 766efa332..2e07da256 100644
--- a/sentence_transformers/models/CLIPModel.py
+++ b/sentence_transformers/models/CLIPModel.py
@@ -1,7 +1,7 @@
 from torch import nn
 import transformers
 import torch
-from PIL import ImageFile
+from PIL import Image
 
 
 class CLIPModel(nn.Module):
@@ -56,7 +56,7 @@ def tokenize(self, texts):
         image_text_info = []
 
         for idx, data in enumerate(texts):
-            if isinstance(data, ImageFile.ImageFile):  # An Image
+            if isinstance(data, Image.Image):  # An Image
                 images.append(data)
                 image_text_info.append(0)
             else:  # A text