Upload folder using huggingface_hub

README.md

@@ -1,18 +1,18 @@
 ## Usage
+This is a model I trained to mimic a JCM 800 AMP. It doesn't sound very good, but as a first pass, I'm glad I have it.
+
 
 ```python
 import torch
-from model import MetalAmpNet  # make sure model.py is in the working directory
+from model import MetalAmpNet
 
-# Load checkpoint
 ckpt = torch.load("metal_amp_v2.pt", map_location="cpu")
-
-# Initialize model
 model = MetalAmpNet()
 model.load_state_dict(ckpt["model"])
 model.eval()
-
-# Example inference
-# x: torch.Tensor of shape [1, 1, window_size] (single channel)
-# y = model(x)
 ```
+
+
+## COMING SOON
+You can look at `infer.py`, `model.py` and `config.py` for deep-dive.
+![](infer.PNG)