update

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
 *.jpg filter=lfs diff=lfs merge=lfs -text
+checkpoints/ootd/vae/diffusion_pytorch_model.bin filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,14 @@
+---
+title: OOTDiffusion Try-On
+emoji: 👗
+colorFrom: purple
+colorTo: pink
+sdk: gradio
+sdk_version: 4.16.0
+app_file: ./run/gradio_ootd.py
+pinned: false
+---
+
 # OOTDiffusion
 This repository is the official implementation of OOTDiffusion
 

ootd/inference_ootd_hd.py

@@ -25,9 +25,9 @@ from transformers import AutoProcessor, CLIPVisionModelWithProjection
 from transformers import CLIPTextModel, CLIPTokenizer
 
 VIT_PATH = "../checkpoints/clip-vit-large-patch14"
-VAE_PATH = "../checkpoints/ootd"
-UNET_PATH = "../checkpoints/ootd/ootd_hd/checkpoint-36000"
-MODEL_PATH = "../checkpoints/ootd"
+VAE_PATH = "/home/user/app/checkpoints/ootd"
+UNET_PATH = "/home/user/app/checkpoints/ootd/ootd_hd/checkpoint-36000"
+MODEL_PATH = "/home/user/app/checkpoints/ootd"
 
 class OOTDiffusionHD:
 

requirements.txt

@@ -11,4 +11,8 @@ tqdm==4.64.1
 gradio==4.16.0
 config==0.5.1
 einops==0.7.0
-onnxruntime==1.16.2
+onnxruntime==1.16.2
+huggingface_hub==0.25.0 
+torchvision==0.14.0
+torch
+basicsr