Update Visual-CoT demo - 2025-10-12 22:22:10

Fixes:
- Fix LLaVA config registration error (compatibility with newer transformers)
- Update Gradio to latest version (security fixes)
- Auto-deployed via update script

llava/model/builder.py

@@ -56,7 +56,8 @@ def load_pretrained_model(
     else:
         kwargs["torch_dtype"] = torch.bfloat16
 
-    if "llava" in model_name.lower():
+    # Check if model is LLaVA-based (including VisCoT which is built on LLaVA)
+    if "llava" in model_name.lower() or "viscot" in model_name.lower():
         # Load LLaVA model
 
         if "lora" in model_name.lower() and model_base is not None:
@@ -73,7 +74,7 @@ def load_pretrained_model(
 
     image_processor = None
 
-    if "llava" in model_name.lower():
+    if "llava" in model_name.lower() or "viscot" in model_name.lower():
         mm_use_im_start_end = getattr(model.config, "mm_use_im_start_end", False)
         mm_use_im_patch_token = getattr(model.config, "mm_use_im_patch_token", True)
         if mm_use_im_patch_token:

requirements.txt

@@ -8,8 +8,8 @@ transformers==4.37.2
 tokenizers==0.15.1
 sentencepiece==0.1.99
 
-# Gradio and Spaces (use stable version compatible with Spaces)
-gradio==4.36.0  # Stable version without breaking changes
+# Gradio and Spaces (use latest version)
+gradio  # Latest version with all security updates
 spaces>=0.19.4
 
 # Model dependencies