Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -188,7 +188,7 @@ tag_model = ram(pretrained='preset/models/ram_swin_large_14m.pth',
|
|
| 188 |
tag_model.eval()
|
| 189 |
tag_model.to(device, dtype=weight_dtype)
|
| 190 |
|
| 191 |
-
@spaces.GPU(
|
| 192 |
def process(
|
| 193 |
input_image: Image.Image,
|
| 194 |
user_prompt: str,
|
|
@@ -210,8 +210,6 @@ def process(
|
|
| 210 |
resize_preproc = transforms.Compose([
|
| 211 |
transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
|
| 212 |
])
|
| 213 |
-
|
| 214 |
-
|
| 215 |
user_prompt = _generate_vlm_prompt(
|
| 216 |
vlm_model=vlm_model,
|
| 217 |
vlm_processor=vlm_processor,
|
|
|
|
| 188 |
tag_model.eval()
|
| 189 |
tag_model.to(device, dtype=weight_dtype)
|
| 190 |
|
| 191 |
+
@spaces.GPU()
|
| 192 |
def process(
|
| 193 |
input_image: Image.Image,
|
| 194 |
user_prompt: str,
|
|
|
|
| 210 |
resize_preproc = transforms.Compose([
|
| 211 |
transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
|
| 212 |
])
|
|
|
|
|
|
|
| 213 |
user_prompt = _generate_vlm_prompt(
|
| 214 |
vlm_model=vlm_model,
|
| 215 |
vlm_processor=vlm_processor,
|