Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from PIL import Image
@@ -21,16 +22,18 @@ model.gradient_checkpointing_enable()
 
 processor = AutoProcessor.from_pretrained(model_id)
 
-FastVisionModel.for_inference(model)
-print("กำลังโหลด tokenizer...")
-base_model, tokenizer = FastVisionModel.from_pretrained(
-    "unsloth/Llama-3.2-11B-Vision-Instruct",
-    # load_in_4bit = True,
-    use_gradient_checkpointing = "unsloth",
-)
-
+@spaces.GPU(duration=120)
 # Function to process the image and generate the description
 def generate_description(image: Image.Image, instruction: str):
+
+    FastVisionModel.for_inference(model)
+    print("กำลังโหลด tokenizer...")
+    base_model, tokenizer = FastVisionModel.from_pretrained(
+        "unsloth/Llama-3.2-11B-Vision-Instruct",
+        # load_in_4bit = True,
+        use_gradient_checkpointing = "unsloth",
+    )
+    
     image = image.convert("RGB")
     # image = Resize((224, 224))(image)