Update app.py

app.py

@@ -26,6 +26,7 @@ tokenizer = None
 model = None
 
 # --- Load Models and Tokenizers Function ---
+@spaces.GPU
 def load_model_and_tokenizer():
     """
     Loads the language model and tokenizer from Hugging Face Hub or a local path.
@@ -61,6 +62,7 @@ def load_model_and_tokenizer():
 
 
 # --- Generate Response Function ---
+@spaces.GPU
 def generate_response(
     message: str, # Current user message
     history: list # Gradio Chatbot history format (list of dictionaries with 'role' and 'content')