Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from wtpsplit import SaT
 import json
@@ -6,6 +7,7 @@ import json
 sat = SaT("sat-12l-sm")
 sat.half().to("cuda")
 
+@spaces.GPU(duration=59)
 def segment_text(input_text, multi_doc_input):
     results = {}