diff --git a/predict.py b/predict.py index 8e38b35..93e26e9 100644 --- a/predict.py +++ b/predict.py @@ -504,7 +504,7 @@ class Predictor(BasePredictor): except (FileNotFoundError, UnidentifiedImageError): logging.debug("Bad image prompt; ignoring") # Is there a better way to input images? else: - self.image_prompt = Image.open(image_prompt) + self.image_prompt = str(image_prompt) self.args.both_stages = both_stages for file in self.run():