Spaces:
Runtime error
Runtime error
fix vlm return
Browse files- app/business_logic.py +1 -1
app/business_logic.py
CHANGED
|
@@ -489,7 +489,7 @@ def vlm_auto_generate(image_target_state, image_reference_state, mask_target_sta
|
|
| 489 |
|
| 490 |
if VLM_PROCESSOR is None or VLM_MODEL is None:
|
| 491 |
gr.Warning("Please enable vlm for prompt first")
|
| 492 |
-
return
|
| 493 |
|
| 494 |
messages = construct_vlm_gen_prompt(image_target_state, image_reference_state, mask_target_state, custmization_mode)
|
| 495 |
output_text = run_vlm(VLM_PROCESSOR, VLM_MODEL, messages, device=device)
|
|
|
|
| 489 |
|
| 490 |
if VLM_PROCESSOR is None or VLM_MODEL is None:
|
| 491 |
gr.Warning("Please enable vlm for prompt first")
|
| 492 |
+
return None
|
| 493 |
|
| 494 |
messages = construct_vlm_gen_prompt(image_target_state, image_reference_state, mask_target_state, custmization_mode)
|
| 495 |
output_text = run_vlm(VLM_PROCESSOR, VLM_MODEL, messages, device=device)
|