alexnasa commited on
Commit
dc725ea
·
verified ·
1 Parent(s): 4deb9f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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(duration=120)
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,