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