upload

README.md

@@ -1,5 +1,5 @@
 ---
-title: Diffusion Models Trained with Large Data Are Transferable Visual Models
+title: GenPercept
 emoji: ⚡
 colorFrom: indigo
 colorTo: red

app.py

@@ -27,8 +27,6 @@ import functools
 import os
 import tempfile
 import warnings
-import sys
-sys.path.append("../")
 
 import gradio as gr
 import numpy as np
@@ -273,8 +271,8 @@ def main():
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
-    vae = AutoencoderKL.from_pretrained("./", subfolder='vae')
-    unet = UNet2DConditionModel.from_pretrained('./', subfolder="unet")
+    vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae')
+    unet = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet")
     empty_text_embed = torch.from_numpy(np.load("./empty_text_embed.npy")).to(device, dtype)[None] # [1, 77, 1024]
     
     pipe = GenPerceptPipeline(vae=vae,