add spaces

app.py

@@ -1,3 +1,4 @@
+import spaces
 from huggingface_hub import InferenceClient
 import gradio as gr
 import random
@@ -16,6 +17,8 @@ def format_prompt(message, history):
   prompt += f"[INST] {message} [/INST]"
   return prompt
 
+
+@spaces.GPU()
 def generate(prompt, history, temperature=0.9, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
     temperature = float(temperature)
     if temperature < 1e-2: