timm
PyTorch
medical
Image Feature Extraction
egeozsoy commited on
Commit
7692a27
1 Parent(s): 8050e5e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -80,7 +80,7 @@ def load_model_from_huggingface(repo_id, model_filename):
80
  return model, loading
81
 
82
 
83
- image_paths = sorted(Path('demo_images').glob('*.png')) # TODO replace with image pass
84
  images = torch.stack([process_single_image(image_path) for image_path in image_paths])
85
 
86
  device = "cuda"
 
80
  return model, loading
81
 
82
 
83
+ image_paths = sorted(Path('demo_images').glob('*.png')) # TODO replace with image path
84
  images = torch.stack([process_single_image(image_path) for image_path in image_paths])
85
 
86
  device = "cuda"