Update app.py

app.py

@@ -35,6 +35,7 @@ def create_chat_template_messages(history, prompt):
     return messages
 
 # Async function for generating responses using two models
+@spaces.GPU
 async def generate_responses(prompt, history):
     # Create messages array for chat history and apply template
     messages = create_chat_template_messages(history, prompt)