Spaces:
Runtime error
Runtime error
Li
commited on
Commit
·
706df4b
1
Parent(s):
1bddbe2
update app.py
Browse files
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.
|
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))
|