Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import time
+import spaces
 
 import gradio as gr
 import argparse
@@ -72,7 +73,7 @@ def highlight_text(text, text_list,color="black"):
 
     return result
 
-
+@spaces.GPU
 def warmup(model):
     conv = get_conversation_template(args.model_type)
 
@@ -92,7 +93,7 @@ def warmup(model):
     input_ids = torch.as_tensor(input_ids).cuda()
     for output_ids in model.ea_generate(input_ids):
         ol=output_ids.shape[1]
-
+@spaces.GPU
 def bot(history, temperature, top_p, use_EaInfer, highlight_EaInfer,session_state,):
     if not history:
         return history, "0.00 tokens/s", "0.00", session_state