Update utils.py

utils.py

@@ -3,6 +3,7 @@ import os
 import torch
 from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration
 from functools import lru_cache
+import spaces
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
@@ -23,7 +24,8 @@ ARCHITECTURE_MAP = {
     "AutoModelForCausalLM": AutoModelForCausalLM
 }
 
-# Function to get the model summary with caching
+# Function to get the model summary with caching and GPU support
+@spaces.GPU
 @lru_cache(maxsize=10)
 def get_model_summary(model_name):
     """