Spaces:
Runtime error
Runtime error
Commit
•
f838b34
1
Parent(s):
04ca745
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from wtpsplit import SaT
|
| 3 |
import json
|
|
@@ -6,6 +7,7 @@ import json
|
|
| 6 |
sat = SaT("sat-12l-sm")
|
| 7 |
sat.half().to("cuda")
|
| 8 |
|
|
|
|
| 9 |
def segment_text(input_text, multi_doc_input):
|
| 10 |
results = {}
|
| 11 |
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
from wtpsplit import SaT
|
| 4 |
import json
|
|
|
|
| 7 |
sat = SaT("sat-12l-sm")
|
| 8 |
sat.half().to("cuda")
|
| 9 |
|
| 10 |
+
@spaces.GPU(duration=59)
|
| 11 |
def segment_text(input_text, multi_doc_input):
|
| 12 |
results = {}
|
| 13 |
|