fixing

app.py

@@ -1,6 +1,6 @@
 import os
 import torch
-from transformers import AutoModelForVision2Seq, AutoProcessor
+from transformers import AutoModelForCausalLM, AutoProcessor
 from PIL import Image
 import gradio as gr
 
@@ -15,7 +15,14 @@ def load_model():
     hub_model_path = "Aekanun/thai-handwriting-llm"
     
     processor = AutoProcessor.from_pretrained(base_model_path, token=token)
-    model = AutoModelForVision2Seq.from_pretrained(hub_model_path, token=token)
+    
+    # โหลดโมเดลด้วย CausalLM ตาม task_type ใน adapter_config
+    model = AutoModelForCausalLM.from_pretrained(
+        hub_model_path,
+        trust_remote_code=True,
+        target_modules=["q_proj", "v_proj"],  # จาก adapter_config
+        token=token
+    )
     
     return model, processor