Revert back to 8B

app.py

@@ -184,12 +184,12 @@ if is_running_on_spaces():
     # Load from Hugging Face Hub
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",
-        filename="diffusion-model-3B.pth",
+        filename="diffusion-model-8B.pth",
         token=os.getenv("HF_TOKEN")
     )
 else:
     # Load from local path
-    ckpt_path = "diffusion-model-3B.pth"  # change to your actual local path
+    ckpt_path = "diffusion-model-8B.pth"  # change to your actual local path
 
 model, tokenizer = load_trained_model(checkpoint_path=ckpt_path)
 print("✅ Model loaded.")