Update app.py

app.py

@@ -7,27 +7,40 @@ import io
 @st.cache_resource
 def load_model():
     model_name = "Qwen/Qwen2-VL-7B-Instruct"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
-    return tokenizer, model
+    try:
+        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:
+        st.error(f"Ошибка при загрузке модели: {str(e)}")
+        return None, None
 
 def generate_response(prompt, image, tokenizer, model):
-    if image:
-        image = Image.open(image).convert('RGB')
-        inputs = tokenizer.from_pretrained(prompt, images=[image], return_tensors='pt').to(model.device)
-    else:
-        inputs = tokenizer(prompt, return_tensors='pt').to(model.device)
-    
-    with torch.no_grad():
-        outputs = model.generate(**inputs, max_new_tokens=100)
+    if tokenizer is None or model is None:
+        return "Модель не загружена. Пожалуйста, проверьте ошибки выше."
     
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response
+    try:
+        if image:
+            image = Image.open(image).convert('RGB')
+            inputs = tokenizer.from_pretrained(prompt, images=[image], return_tensors='pt').to(model.device)
+        else:
+            inputs = tokenizer(prompt, return_tensors='pt').to(model.device)
+        
+        with torch.no_grad():
+            outputs = model.generate(**inputs, max_new_tokens=100)
+        
+        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        return response
+    except Exception as e:
+        return f"Ошибка при генерации ответа: {str(e)}"
 
 st.title("Чат с Qwen VL-7B-Instruct")
 
 tokenizer, model = load_model()
 
+if tokenizer is None or model is None:
+    st.warning("Модель не загружена. Приложение может работать некорректно.")
+
 if "messages" not in st.session_state:
     st.session_state.messages = []