Change model

app.py

@@ -27,10 +27,9 @@ token_probabilities = np.array([token_probs_dict[str(i)] for i in range(len(toke
 
 def load_model():
     ckpt_path = hf_hub_download(
-        repo_id="ruurd/tini_model",
+        repo_id="ruurd/tini_bi_m",
         filename="diffusion-model.pth",
-        token=os.getenv("HF_TOKEN"),
-        revision="8bb2d44"
+        token=os.getenv("HF_TOKEN")
     )
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -127,9 +126,6 @@ def confidence_guided_noising(input_ids, answer_start, confidences, noise_clippi
 
     return noised
 
-
-
-
 @spaces.GPU
 def generate_diffusion_text(input_ids):
     with torch.no_grad():