Update app.py

app.py

@@ -1,15 +1,19 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM
-from transformers.models.qwen2.tokenization_qwen2 import Qwen2Tokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from PIL import Image
 import io
+import importlib
+
+def check_transformers_version():
+    import transformers
+    return transformers.__version__
 
 @st.cache_resource
 def load_model():
     model_name = "Qwen/Qwen2-VL-7B-Instruct"
     try:
-        tokenizer = Qwen2Tokenizer.from_pretrained(model_name, trust_remote_code=True)
+        tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
         model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
         return tokenizer, model
     except Exception as e:
@@ -37,10 +41,14 @@ def generate_response(prompt, image, tokenizer, model):
 
 st.title("Чат с Qwen VL-7B-Instruct")
 
+transformers_version = check_transformers_version()
+st.info(f"Версия transformers: {transformers_version}")
+
 tokenizer, model = load_model()
 
 if tokenizer is None or model is None:
     st.warning("Модель не загружена. Приложение может работать некорректно.")
+    st.info("Попробуйте установить последнюю версию transformers: pip install transformers --upgrade")
 else:
     st.success("Модель успешно загружена!")