Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import transformers
 import os
 from PIL import Image
+import spaces
 
 def process_vision_info(messages):
     image_inputs = []
@@ -87,6 +88,7 @@ def visualize_text_token_probabilities(text: str):
         print(f"An error occurred during text processing: {e}")
         return [(f"An error occurred: {str(e)}", None)]
 
+@spaces.GPU
 def generate_and_visualize_vl_probabilities(image, prompt: str):
     if not vl_model_loaded:
         return [(f"Vision-Language Model failed to load: {vl_model_error}", None)]