test-space

Running on Zero

vikhyatk commited on May 27

Commit

63cdb57

verified ·

1 Parent(s): aa8a8fc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ from PIL import ImageDraw
 from torchvision.transforms.v2 import Resize
 os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
-tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream-next", revision=REVISION)
 moondream = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream-next",
     trust_remote_code=True,

 from torchvision.transforms.v2 import Resize
 os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
+tokenizer = AutoTokenizer.from_pretrained("moondream/starmie-v1")
 moondream = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream-next",
     trust_remote_code=True,