Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|