alexnasa commited on
Commit
d7535db
·
verified ·
1 Parent(s): a2bb6b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -210,6 +210,17 @@ def process(
210
  transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
211
  ])
212
 
 
 
 
 
 
 
 
 
 
 
 
213
  # with torch.no_grad():
214
  seed_everything(seed)
215
  generator = torch.Generator(device=device)
 
210
  transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
211
  ])
212
 
213
+
214
+ prompt_tag = _generate_vlm_prompt(
215
+ vlm_model=vlm_model,
216
+ vlm_processor=vlm_processor,
217
+ process_vision_info=process_vision_info,
218
+ pil_image=input_image,
219
+ device=device,
220
+ )
221
+
222
+ print(f'oh lala, prompt tag:{prompt_tag}')
223
+
224
  # with torch.no_grad():
225
  seed_everything(seed)
226
  generator = torch.Generator(device=device)