Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import spaces
+
+
 
 model = AutoModelForCausalLM.from_pretrained("instruction-pretrain/instruction-synthesizer")
+model.to('cuda')
 tokenizer = AutoTokenizer.from_pretrained("instruction-pretrain/instruction-synthesizer")
 
 def parse_pred(pred):
@@ -39,6 +43,7 @@ def get_instruction_response_pairs(context):
     pred = tokenizer.decode(outputs[pred_start:], skip_special_tokens=True)
     return parse_pred(pred)
 
+@spaces.GPU
 def generate_pairs(context):
     instruction_response_pairs = get_instruction_response_pairs(context)
     output = ""