use the latest checkpoint from hub

app.py

@@ -4,7 +4,7 @@ import random
 import gradio as gr
 import numpy as np
 import torch
-from diffusers import DiffusionPipeline, UNet2DConditionModel
+from diffusers import DiffusionPipeline
 import spaces
 import uuid
 
@@ -18,11 +18,9 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "768"))
 TOKEN = os.getenv("HF_TOKEN")
 
 
-pipe_id = "SPRIGHT-T2I/spright-t2i-v1"
-unet = UNet2DConditionModel.from_pretrained(pipe_id, subfolder="unet_ema", torch_dtype=torch.float16)
+pipe_id = "SPRIGHT-T2I/spright-t2i-sd2"
 pipe = DiffusionPipeline.from_pretrained(
     pipe_id,
-    unet=unet,
     torch_dtype=torch.float16,
     use_safetensors=True,
     token=TOKEN,