Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import spaces
 
 model_id = "tiiuae/falcon-rw-1b"
 
@@ -9,6 +10,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32)
 
 # Chat logic
+@spaces.GPU
 def chat_with_bot(user_input, history):
     history = history or []
     prompt = ""