Update app.py

app.py

@@ -1,11 +1,12 @@
 import streamlit as st
-from transformers import AutoProcessor, Qwen2VLForConditionalGeneration
+from transformers import AutoProcessor, Qwen2VLForConditionalGeneration, AutoConfig
 from PIL import Image
 import torch
 
-# Load the processor and model directly
+# Load the processor and model configuration
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
-model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
+config = AutoConfig.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
+model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", config=config)
 
 # Streamlit app
 st.title("Image Description Generator")