Update code/inference.py

code/inference.py

@@ -72,7 +72,7 @@ def predict_fn(data, model_and_tokenizer):
         image = Image.open(BytesIO(response.content)).convert("RGB")
     elif image_file.startswith("s3://"):
         s3 = boto3.client("s3")
-        s3_path = s3_path[5:]
+        s3_path = image_file[5:]
         bucket = s3_path.split('/')[0]
         s3_key = '/'.join(s3_path.split('/')[1:])
         with tempfile.NamedTemporaryFile() as temp_file: