Update app.py

app.py

@@ -50,13 +50,15 @@ if IN_SPACES:
     )
 
 os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
-tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream-next")
+revision = "7b92e6646dd60437d698a2fa75c7f65a90320dad"
+tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream-next", revision=revision)
 moondream = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream-next",
     trust_remote_code=True,
     torch_dtype=torch.float16,
     device_map={"": "cuda"},
     attn_implementation="flash_attention_2",
+    revision=revision
 )
 
 # CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]