Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoProcessor, AutoModelForCausalLM
+from transformers import AutoConfig, AutoProcessor, AutoModelForCausalLM
 import spaces
 from PIL import Image 
 import subprocess
@@ -11,6 +11,11 @@ from io import BytesIO
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
+config = AutoConfig.from_pretrained('medieval-data/florence2-medieval-bbox-line-detection')
+    # Modify the vision configuration
+if hasattr(config, 'vision_config'):
+    config.vision_config.model_type = 'davit'
+    
 model = AutoModelForCausalLM.from_pretrained('medieval-data/florence2-medieval-bbox-line-detection', trust_remote_code=True).eval()
 processor = AutoProcessor.from_pretrained('medieval-data/florence2-medieval-bbox-line-detection', trust_remote_code=True)