Update code/inference.py
Browse files- code/inference.py +1 -1
code/inference.py
CHANGED
@@ -77,7 +77,7 @@ def predict_fn(data, model_and_tokenizer):
|
|
77 |
s3_key = '/'.join(s3_path.split('/')[1:])
|
78 |
with tempfile.NamedTemporaryFile() as temp_file:
|
79 |
s3.download_file(bucket, s3_key, temp_file.name)
|
80 |
-
|
81 |
else:
|
82 |
image = Image.open(image_file).convert("RGB")
|
83 |
|
|
|
77 |
s3_key = '/'.join(s3_path.split('/')[1:])
|
78 |
with tempfile.NamedTemporaryFile() as temp_file:
|
79 |
s3.download_file(bucket, s3_key, temp_file.name)
|
80 |
+
image = Image.open(temp_file.name).convert("RGB")
|
81 |
else:
|
82 |
image = Image.open(image_file).convert("RGB")
|
83 |
|