Update app.py

app.py

@@ -3,6 +3,8 @@ from transformers import AutoModel, AutoTokenizer
 import torch
 import torchvision.transforms as T
 from torchvision.transforms.functional import InterpolationMode
+from Models.modeling_llavaqw import LlavaQwModel
+
 
 
 
@@ -12,7 +14,7 @@ IMAGENET_STD = (0.229, 0.224, 0.225)
 
 model_name = "torettomarui/Llava-qw"  
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, use_fast=False)
-model = AutoModel.from_pretrained(
+model = LlavaQwModel.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,