Update app.py

Use huggingface model hub instead to store checkpoints instead of including them directly.

app.py

@@ -6,6 +6,7 @@ import os
 import random
 from mapping import reduced_genre_mapping, reduced_style_mapping, reverse_reduced_genre_mapping, reverse_reduced_style_mapping
 from diffusers import AutoencoderKL
+from huggingface_hub import hf_hub_download
 from models.DiT import DiT
 
 # Global settings 
@@ -19,7 +20,11 @@ global_progress = 0
 
 def load_dit_model(dit_size):
     """Load DiT model of specified size"""
-    ckpt_path = f"./ckpts/DiT_{dit_size}_final.pth"
+    #ckpt_path = f"./ckpts/DiT_{dit_size}_final.pth"
+    ckpt_path = hf_hub_download(
+        repo_id = "kaupane/DiT-Wikiart",
+        filename = f"DiT_{dit_size}_final.pth"
+    )
     if not os.path.exists(ckpt_path):
         raise FileNotFoundError(f"Checkpoint not found at {ckpt_path}")