Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain.prompts import HumanMessagePromptTemplate, ChatPromptTemplate
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
@@ -9,6 +10,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Define the Langchain chatbot function
+@spaces.GPU
 def chatbot(message, history):
     # Create a Langchain prompt template
     prompt_template = HumanMessagePromptTemplate.from_message(message)
@@ -35,4 +37,4 @@ with gr.Blocks() as demo:
     )
 
     # Launch the Gradio app
-    demo.launch()
+demo.launch()