Upload README.md with huggingface_hub

README.md

@@ -1,40 +1,8 @@
----
-tags:
-- unet
-- pix2pix
-- pytorch
-library_name: pytorch
-license: wtfpl
-datasets:
-- K00B404/pix2pix_flux_set
-language:
-- en
-pipeline_tag: image-to-image
----
-
 # Pix2Pix UNet Model
 
-## Model Description
-Custom UNet model for Pix2Pix image translation.
 - **Image Size:** 256
 - **Model Type:** small_UNet (256)
-
-## Usage
-
-```python
-import torch
-from small_256_model import UNet as small_UNet
-from big_1024_model import UNet as big_UNet
-big = True
-# Load the model
-name='big_model_weights.pth' if big else 'small_model_weights.pth'
-checkpoint = torch.load(name)
-model = big_UNet() if checkpoint['model_config']['big'] else small_UNet()
-model.load_state_dict(checkpoint['model_state_dict'])
-model.eval()
-
-Model Architecture
-
+## Model Architecture
 UNet(
   (encoder): Sequential(
     (0): Conv2d(3, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
@@ -52,4 +20,4 @@ UNet(
     (4): ConvTranspose2d(64, 3, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
     (5): Tanh()
   )
-) 
+)