Also added decorator for text probs

app.py

@@ -52,6 +52,7 @@ except Exception as e:
     print(f"Error loading Vision-Language model: {vl_model_error}")
     vl_model, vl_processor = None, None
 
+@spaces.GPU
 def visualize_text_token_probabilities(text: str):
     if not text_model_loaded:
         return [(f"Text Model failed to load: {text_model_error}", None)]