Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from pathlib import Path
 import pandas as pd
+import spaces
 
 model_checkpoint = "HuggingFaceTB/SmolLM-1.7B"
 model = AutoModelForCausalLM.from_pretrained(model_checkpoint)
@@ -48,7 +49,7 @@ with gr.Blocks() as demo:
                 chatbot1 = gr.Chatbot()
             msg = gr.Textbox()
             clear = gr.ClearButton([msg, chatbot])
-
+            @spaces.GPU(duration=200)
             def respond(message, chat_history):
                 response = pipe(message)
                 bot_message = response[0]["generated_text"]