Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+import spaces
 
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("Yoxas/autotrain-phi3-statistical", trust_remote_code=True)
@@ -9,6 +10,7 @@ model = AutoModelForCausalLM.from_pretrained("Yoxas/autotrain-phi3-statistical",
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code=True)
 
 # Define the chatbot function
+@spaces.GPU(duration=120)
 def chatbot(input_text):
     response = pipe(input_text, max_length=150, num_return_sequences=1)
     return response[0]['generated_text']