rev app

app.py

@@ -143,8 +143,6 @@ def load_model():
             torch_dtype=torch.float16
         ).to('cuda')
 
-        FastVisionModel.for_inference(model) ###ลองแก้ไขปัญหา torch
-
         print("โหลดโมเดลสำเร็จ!")
         return True
         
@@ -164,6 +162,9 @@ def process_image(image):
 
     ### โหลด tokenizer จาก base model
     from unsloth import FastVisionModel
+
+    FastVisionModel.for_inference(model) ###ลองแก้ไขปัญหา torch
+
     from transformers import AutoTokenizer
     print("กำลังโหลด tokenizer...")
     base_model, tokenizer = FastVisionModel.from_pretrained(