ZeroGPU (#1)

- ZeroGPU (5a92cfbc3ebb6de4fa40e6be3f1f006dd40f7bb7)


Co-authored-by: Charles Bensimon <[email protected]>

app.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import spaces
 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)
 
 
+@spaces.GPU(duration=10)
 def answer_question(image, question):
     yield "Encoding image..."