Spaces:
Runtime error
Runtime error
fix edge case saving with no images
Browse files- app_dialogue.py +12 -2
app_dialogue.py
CHANGED
|
@@ -258,7 +258,12 @@ def flag_dope(
|
|
| 258 |
images.append(ex[0]["file"])
|
| 259 |
prev_ex_is_image = True
|
| 260 |
|
| 261 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 262 |
dope_dataset_writer.flag(
|
| 263 |
flag_data=[
|
| 264 |
model_selector,
|
|
@@ -287,7 +292,12 @@ def flag_problematic(
|
|
| 287 |
if isinstance(ex[0], dict):
|
| 288 |
images.append(ex[0]["file"])
|
| 289 |
|
| 290 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 291 |
problematic_dataset_writer.flag(
|
| 292 |
flag_data=[
|
| 293 |
model_selector,
|
|
|
|
| 258 |
images.append(ex[0]["file"])
|
| 259 |
prev_ex_is_image = True
|
| 260 |
|
| 261 |
+
if len(images)== 0:
|
| 262 |
+
black_image = Image.new('RGB', (20, 20), (0, 0, 0))
|
| 263 |
+
black_image.save("/tmp/gradio/fake_image.png")
|
| 264 |
+
image_flag = {'path': "/tmp/gradio/fake_image.png", 'size': None, 'orig_name': None, 'mime_type': 'image/png', 'is_stream': False, 'meta': {'_type': 'gradio.FileData'}}
|
| 265 |
+
else:
|
| 266 |
+
image_flag = images[0]
|
| 267 |
dope_dataset_writer.flag(
|
| 268 |
flag_data=[
|
| 269 |
model_selector,
|
|
|
|
| 292 |
if isinstance(ex[0], dict):
|
| 293 |
images.append(ex[0]["file"])
|
| 294 |
|
| 295 |
+
if len(images)== 0:
|
| 296 |
+
black_image = Image.new('RGB', (20, 20), (0, 0, 0))
|
| 297 |
+
black_image.save("/tmp/gradio/fake_image.png")
|
| 298 |
+
image_flag = {'path': "/tmp/gradio/fake_image.png", 'size': None, 'orig_name': None, 'mime_type': 'image/png', 'is_stream': False, 'meta': {'_type': 'gradio.FileData'}}
|
| 299 |
+
else:
|
| 300 |
+
image_flag = images[0]
|
| 301 |
problematic_dataset_writer.flag(
|
| 302 |
flag_data=[
|
| 303 |
model_selector,
|