mattia-re-learn commited on
Commit
b46c431
·
verified ·
1 Parent(s): f808930

Update code/inference.py

Browse files
Files changed (1) hide show
  1. 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
- image = Image.open(temp_file).convert("RGB")
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