Csplk commited on
Commit
087a962
ยท
verified ยท
1 Parent(s): 6c84d1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,13 +34,13 @@ from torchvision.transforms.v2 import Resize
34
  #moondream.eval()
35
 
36
  """Load Moondream model and tokenizer."""
37
- tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
38
  moondream = AutoModelForCausalLM.from_pretrained(
39
  "vikhyatk/moondream2",
40
  revision="2025-01-09",
41
  trust_remote_code=True,
42
  device_map={"": "cuda"},
43
  )
 
44
 
45
  @spaces.GPU(durtion="150")
46
  def answer_questions(image_tuples, prompt_text):
 
34
  #moondream.eval()
35
 
36
  """Load Moondream model and tokenizer."""
 
37
  moondream = AutoModelForCausalLM.from_pretrained(
38
  "vikhyatk/moondream2",
39
  revision="2025-01-09",
40
  trust_remote_code=True,
41
  device_map={"": "cuda"},
42
  )
43
+ tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
44
 
45
  @spaces.GPU(durtion="150")
46
  def answer_questions(image_tuples, prompt_text):