Update README.md
Browse files
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
|
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"
|