Spaces:

HuggingFaceM4
/

ai_raven

Runtime error

VictorSanh commited on Mar 10

Commit

12089a1

•

1 Parent(s): f20057b

add community grant

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,6 +6,7 @@ from PIL import Image
 import imagehash
 import cv2
 import os
 from transformers import AutoProcessor, AutoModelForCausalLM
 from transformers.image_utils import to_numpy_array, PILImageResampling, ChannelDimension
@@ -117,7 +118,7 @@ def load_sample():
     return sample["image"], sample["label"], "", "", ""
-# @spaces.GPU(duration=180)
 def model_inference(
     image,
 ):

 import imagehash
 import cv2
 import os
+import spaces
 from transformers import AutoProcessor, AutoModelForCausalLM
 from transformers.image_utils import to_numpy_array, PILImageResampling, ChannelDimension
     return sample["image"], sample["label"], "", "", ""
+@spaces.GPU(duration=180)
 def model_inference(
     image,
 ):