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,
|