Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import sys
@@ -47,6 +48,7 @@ def convert_history(chat_history, max_input_length=1024):
     return history_text
 
 @torch.inference_mode()
+@spaces.GPU
 def instruct(instruction, max_token_output=1024):
     input_text = instruction
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)