quantumiracle-git commited on
Commit
ede8743
·
1 Parent(s): 4e68dcf

Update hfserver.py

Browse files
Files changed (1) hide show
  1. hfserver.py +2 -1
hfserver.py CHANGED
@@ -380,7 +380,8 @@ class HuggingFaceDatasetSaver(FlaggingCallback):
380
  self.dataset_dir,
381
  utils.strip_invalid_filename_characters(component.label),
382
  )
383
- filepath = component.deserialize(sample, save_dir, None)
 
384
  csv_data.append(filepath)
385
  if isinstance(component, tuple(file_preview_types)):
386
  csv_data.append(
 
380
  self.dataset_dir,
381
  utils.strip_invalid_filename_characters(component.label),
382
  )
383
+ # filepath = component.deserialize(sample, save_dir, None)
384
+ filepath = component.deserialize(sample, save_dir, None) if sample is not None and str(component)!='image' else "" # not save image
385
  csv_data.append(filepath)
386
  if isinstance(component, tuple(file_preview_types)):
387
  csv_data.append(