Update app.py

app.py

@@ -1,6 +1,6 @@
 import spaces
 import gradio as gr
-from transformers import AutoTokenizer, PreTrainedTokenizer, PreTrainedTokenizerFast, LlavaLlamaForCausalLM, TextIteratorStreamer
+from transformers import AutoTokenizer, PreTrainedTokenizer, PreTrainedTokenizerFast, AutoModelForCausalLM, TextIteratorStreamer
 import torch
 import torch.amp.autocast_mode
 from PIL import Image
@@ -19,8 +19,8 @@ PLACEHOLDER = "Merhaba! Size nasıl yardımcı olabilirim? Bir resim yükleyip h
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, use_fast=True)
 assert isinstance(tokenizer, PreTrainedTokenizer) or isinstance(tokenizer, PreTrainedTokenizerFast), f"Expected PreTrainedTokenizer, got {type(tokenizer)}"
 
-model = LlavaLlamaForCausalLM.from_pretrained(MODEL_PATH, torch_dtype="bfloat16", device_map=0)
-assert isinstance(model, LlavaLlamaForCausalLM), f"Expected LlavaLlamaForCausalLM, got {type(model)}"
+model = AutoModelForCausalLM.from_pretrained(MODEL_PATH, torch_dtype="bfloat16", device_map=0)
+# assert isinstance(model, LlavaLlamaForCausalLM), f"Expected LlavaLlamaForCausalLM, got {type(model)}"
 
 
 @spaces.GPU()