Spaces:

sayantan47
/

Image-Captioner-using-Moondream-2

Sleeping

sayantan47 commited on Jul 15

Commit

85adc50

verified ·

1 Parent(s): 3f68a21

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ model = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream2",
     revision="2025-06-21",
     trust_remote_code=True,
-    device_map={"": "device"}  # or "cpu" / "mps"
 )
 tokenizer = AutoTokenizer.from_pretrained(

     "vikhyatk/moondream2",
     revision="2025-06-21",
     trust_remote_code=True,
+    device_map={"": device}  # or "cpu" / "mps"
 )
 tokenizer = AutoTokenizer.from_pretrained(