chore: restructure readme, folders

README.md

@@ -10,5 +10,9 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Reference 
+- [Steganography](https://github.com/kelvins/steganography) 
+- [Watermarking Images in Self-Supervised Latent-Spaces](https://arxiv.org/pdf/2112.09581.pdf)
+    * [Github](https://github.com/facebookresearch/ssl_watermarking) 
+    * [HuggingFace Space](https://huggingface.co/spaces/pierrefdz/ssl_watermarking)
+    * [Explanation](https://pierrefdz.github.io/publications/sslwatermarking/)

SSL_watermark.py

@@ -4,8 +4,8 @@ import json
 import torch
 from torchvision import transforms
 
-import torch_utils 
-import image_utils
+import util_functions.torch_utils as torch_utils 
+import util_functions.image_utils as image_utils
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -13,8 +13,8 @@ torch.manual_seed(0)
 np.random.seed(0)
 
 print('Building backbone and normalization layer...')
-backbone = torch_utils.build_backbone(path='dino_r50.pth')
-normlayer = torch_utils.load_normalization_layer(path='out2048.pth')
+backbone = torch_utils.build_backbone(path='models/dino_r50.pth')
+normlayer = torch_utils.load_normalization_layer(path='models/out2048.pth')
 model = torch_utils.NormLayerWrapper(backbone, normlayer)
 
 print('Building the hypercone...')

app.py

@@ -1,6 +1,6 @@
 import gradio as gr 
 from steganography import Steganography
-from utils import draw_multiple_line_text, generate_qr_code
+from util_functions.utils import draw_multiple_line_text, generate_qr_code
 from SSL_watermark import encode, decode