rev app

app.py

@@ -12,15 +12,18 @@ try:
 except Exception as e:
     print(f"Failed to install packages: {e}")
 
+# [แก้ 1] ย้าย environment variables มาไว้ก่อน imports
 import os
 os.environ['NVIDIA_VISIBLE_DEVICES'] = ''
 
 import warnings
 import torch
-
-torch._dynamo.config.suppress_errors = True 
+torch._dynamo.config.suppress_errors = True
 torch._dynamo.config.verbose = False
 
+# [แก้ 2] ย้าย imports มาไว้ที่ module level
+from unsloth import FastVisionModel
+from transformers import AutoModelForVision2Seq
 from transformers import TextStreamer
 import gradio as gr
 from huggingface_hub import login
@@ -34,16 +37,17 @@ if 'HUGGING_FACE_HUB_TOKEN' in os.environ:
 else:
     print("คำเตือน: ไม่พบ HUGGING_FACE_HUB_TOKEN")
 
-###@spaces.GPU
+# [แก้ 3] เพิ่ม @spaces.GPU decorator
+@spaces.GPU
 def model_context():
     _tokenizer = None
     _model = None
-    
+
     def init_models():
         nonlocal _tokenizer, _model
         try:
             print("กำลังโหลด tokenizer...")
-            from unsloth import FastVisionModel
+            # [แก้ 4] ลบ imports ออกจาก function
             base_model, _tokenizer = FastVisionModel.from_pretrained(
                 "unsloth/Llama-3.2-11B-Vision-Instruct",
                 use_gradient_checkpointing = "unsloth"
@@ -51,7 +55,7 @@ def model_context():
             print("โหลด tokenizer สำเร็จ")
 
             print("กำลังโหลดโมเดล fine-tuned...")
-            from transformers import AutoModelForVision2Seq
+            # [แก้ 5] ลบ import ออกจาก function
             _model = AutoModelForVision2Seq.from_pretrained(
                 "Aekanun/Llama-3.2-11B-Vision-Instruct-XRay",
                 load_in_4bit=True,