Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-
+import spaces
 # Load the model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("Yoxas/autotrain-test")
 model = AutoModelForCausalLM.from_pretrained("Yoxas/autotrain-test")
-
+@spaces.GPU(duration=120)
 def chatbot_response(input_text, max_length, temperature):
     inputs = tokenizer(input_text, return_tensors="pt", padding=True)
     attention_mask = inputs.attention_mask