Spaces:

vikhyatk
/

moondream1

Running on Zero

cbensimon HF staff commited on Feb 7

Commit

5a92cfb

•

1 Parent(s): b77f53d

ZeroGPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
 from __future__ import annotations
 import torch
 from PIL import Image
 from einops import rearrange
@@ -1185,6 +1186,7 @@ def cached_vision_encoder(image):
         return image_vec.to(DEVICE, dtype=DTYPE)
 def answer_question(image, question):
     yield "Encoding image..."

 from __future__ import annotations
+import spaces
 import torch
 from PIL import Image
 from einops import rearrange
         return image_vec.to(DEVICE, dtype=DTYPE)
+@spaces.GPU(duration=10)
 def answer_question(image, question):
     yield "Encoding image..."