Spaces:

Ruurd
/

lad

Running on Zero

Ruurd commited on Apr 8

Commit

b7639f5

1 Parent(s): 9aaa660

Correct model repo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def load_model():
     # 1. Download the checkpoint
     checkpoint_path = hf_hub_download(
-        repo_id="ruurd/diffusion-llama",
         filename="diffusion-model.pth",
         token=os.getenv("HF_TOKEN")
     )

     # 1. Download the checkpoint
     checkpoint_path = hf_hub_download(
+        repo_id="ruurd/tini_model",
         filename="diffusion-model.pth",
         token=os.getenv("HF_TOKEN")
     )