quantumiracle-git
commited on
Commit
•
f18b79d
1
Parent(s):
ede8743
Update hfserver.py
Browse files- hfserver.py +4 -1
hfserver.py
CHANGED
@@ -381,7 +381,10 @@ class HuggingFaceDatasetSaver(FlaggingCallback):
|
|
381 |
utils.strip_invalid_filename_characters(component.label),
|
382 |
)
|
383 |
# filepath = component.deserialize(sample, save_dir, None)
|
384 |
-
|
|
|
|
|
|
|
385 |
csv_data.append(filepath)
|
386 |
if isinstance(component, tuple(file_preview_types)):
|
387 |
csv_data.append(
|
|
|
381 |
utils.strip_invalid_filename_characters(component.label),
|
382 |
)
|
383 |
# filepath = component.deserialize(sample, save_dir, None)
|
384 |
+
if sample is not None and str(component)!='image':
|
385 |
+
filepath = component.deserialize(sample, save_dir, None)
|
386 |
+
else:
|
387 |
+
filepath = component.deserialize(sample, None, None) # not saving image
|
388 |
csv_data.append(filepath)
|
389 |
if isinstance(component, tuple(file_preview_types)):
|
390 |
csv_data.append(
|