Spaces:
Running
on
Zero
Running
on
Zero
Try again
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ token_probabilities = np.array([token_probs_dict[str(i)] for i in range(len(toke
|
|
| 28 |
|
| 29 |
def load_model():
|
| 30 |
ckpt_path = hf_hub_download(
|
| 31 |
-
repo_id="ruurd/
|
| 32 |
filename="diffusion-model.pth",
|
| 33 |
token=os.getenv("HF_TOKEN")
|
| 34 |
)
|
|
|
|
| 28 |
|
| 29 |
def load_model():
|
| 30 |
ckpt_path = hf_hub_download(
|
| 31 |
+
repo_id="ruurd/tini_model",
|
| 32 |
filename="diffusion-model.pth",
|
| 33 |
token=os.getenv("HF_TOKEN")
|
| 34 |
)
|