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 |
|