Correct name of model

app.py

@@ -37,7 +37,7 @@ assistant_marker_ids = tokenizer.encode("<|start_header_id|>assistant<|end_heade
 def load_model():
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",
-        filename="diffusion-model-8B.pth",
+        filename="diffusion-model-8B.pt",
         token=os.getenv("HF_TOKEN"),
         # revision="1ffb916dd34f442f87cf06dda74b96f86eaf1d15",
     )