Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 from transformers import pipeline, AutoTokenizer
 import torch
 import os
-
-
+import spaces
+@spaces.GPU
 def load_model(model_name):
     return pipeline("text-generation", model=model_name, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True, token=os.environ["token"])
-
+@spaces.GPU()
 def generate(
     model_name,
     user_input,