Spaces:

lyttt
/

VLV_Caption

Sleeping

lyttt commited on 29 days ago

Commit

382da0b

1 Parent(s): 762fd7d

.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import numpy as np
 model_name_or_path = "lyttt/VLV_captioner"
 model = AutoModel.from_pretrained(model_name_or_path, revision="master", trust_remote_code=True,low_cpu_mem_usage=False)
-@spaces.GPU(duration=120)
 def greet(image):
     if image.dtype != np.uint8:
         image = (np.clip(image, 0, 1) * 255).astype(np.uint8)

 model_name_or_path = "lyttt/VLV_captioner"
 model = AutoModel.from_pretrained(model_name_or_path, revision="master", trust_remote_code=True,low_cpu_mem_usage=False)
+# @spaces.GPU(duration=120)
 def greet(image):
     if image.dtype != np.uint8:
         image = (np.clip(image, 0, 1) * 255).astype(np.uint8)