k4d3 commited on
Commit
e81d17c
1 Parent(s): 8c06571

Signed-off-by: Balazs Horvath <[email protected]>

Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -290,11 +290,13 @@ pip install timm
290
 
291
  The following inference script for the tagger needs a folder as input, be warned that it also converts WebP images to PNG and you can specify tags to be ignored and some other stuff! I recommend reading through it and changing whatever you need.
292
 
 
 
293
  <div style="background-color: lightyellow; padding: 10px;">
294
  <details>
295
  <summary>Click to reveal inference script</summary>
296
 
297
- ```py
298
  import os
299
  import torch
300
  from torchvision import transforms
 
290
 
291
  The following inference script for the tagger needs a folder as input, be warned that it also converts WebP images to PNG and you can specify tags to be ignored and some other stuff! I recommend reading through it and changing whatever you need.
292
 
293
+ [Colab Notebook](https://colab.research.google.com/drive/1gIB2fGjLAuh6s_hrNlIPCkw_3jodoFP0?usp=sharing)
294
+
295
  <div style="background-color: lightyellow; padding: 10px;">
296
  <details>
297
  <summary>Click to reveal inference script</summary>
298
 
299
+ ```python
300
  import os
301
  import torch
302
  from torchvision import transforms