Li commited on
Commit
706df4b
·
1 Parent(s): 1bddbe2

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def generate(
64
  bad_words_ids = [[b] for b in bad_words_ids]
65
  min_loc_token_id = min(loc_token_ids)
66
  max_loc_token_id = max(loc_token_ids)
67
- image = Image.open(image).convert("RGB")
68
  width = image.width
69
  height = image.height
70
  image = image.resize((224, 224))
 
64
  bad_words_ids = [[b] for b in bad_words_ids]
65
  min_loc_token_id = min(loc_token_ids)
66
  max_loc_token_id = max(loc_token_ids)
67
+ image = Image.convert("RGB")
68
  width = image.width
69
  height = image.height
70
  image = image.resize((224, 224))