fix unsloth import

app.py

@@ -3,7 +3,6 @@ import torch
 from PIL import Image
 import os
 from dotenv import load_dotenv
-from transformers import FastVisionModel
 import spaces
 
 load_dotenv()
@@ -27,6 +26,7 @@ print(f"[Init] Loading model pipeline for the first time: {MODEL_NAME}...")
 @spaces.GPU
 def load_model():
     try:
+        from unsloth import FastVisionModel
         model, processor = FastVisionModel.from_pretrained(
                     MODEL_NAME,
                     device_map="auto",