Csplk commited on
Commit
c589bc2
Β·
verified Β·
1 Parent(s): a1e14ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -21,11 +21,11 @@ import subprocess
21
  # tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
22
  # return model, tokenizer
23
 
24
- """Load Moondream model and tokenizer."""
25
- moondream = AutoModelForCausalLM.from_pretrained(
26
- "vikhyatk/moondream2", trust_remote_code=True, device_map={"": "cuda"}
27
- )
28
- tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
29
 
30
  #model_id = "vikhyatk/moondream2"
31
  #revision = "2025-01-09"
@@ -37,6 +37,13 @@ tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
37
 
38
  #moondream.eval()
39
 
 
 
 
 
 
 
 
40
  @spaces.GPU(durtion="150")
41
  def answer_questions(image_tuples, prompt_text):
42
  result = ""
 
21
  # tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
22
  # return model, tokenizer
23
 
24
+ #"""Load Moondream model and tokenizer."""
25
+ #moondream = AutoModelForCausalLM.from_pretrained(
26
+ # "vikhyatk/moondream2", trust_remote_code=True, device_map={"": "cuda"}
27
+ #)
28
+ #tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
29
 
30
  #model_id = "vikhyatk/moondream2"
31
  #revision = "2025-01-09"
 
37
 
38
  #moondream.eval()
39
 
40
+ model = AutoModelForCausalLM.from_pretrained(
41
+ "vikhyatk/moondream2",
42
+ revision="2025-06-21",
43
+ trust_remote_code=True,
44
+ device_map={"": "cuda"} # ...or 'mps', on Apple Silicon
45
+ )
46
+
47
  @spaces.GPU(durtion="150")
48
  def answer_questions(image_tuples, prompt_text):
49
  result = ""