Spaces:
Running
on
Zero
Running
on
Zero
Update stf_utils.py
Browse files- stf_utils.py +2 -2
stf_utils.py
CHANGED
|
@@ -91,7 +91,7 @@ class STFPipeline:
|
|
| 91 |
self.model = self.load_model()
|
| 92 |
self.template = self.create_template()
|
| 93 |
|
| 94 |
-
@spaces.GPU(duration=
|
| 95 |
def load_model(self):
|
| 96 |
"""모델을 생성하고 GPU에 할당."""
|
| 97 |
model = stf_alternative.create_model(
|
|
@@ -103,7 +103,7 @@ class STFPipeline:
|
|
| 103 |
)
|
| 104 |
return model
|
| 105 |
|
| 106 |
-
@spaces.GPU(duration=
|
| 107 |
def create_template(self):
|
| 108 |
"""템플릿 생성."""
|
| 109 |
template = stf_alternative.Template(
|
|
|
|
| 91 |
self.model = self.load_model()
|
| 92 |
self.template = self.create_template()
|
| 93 |
|
| 94 |
+
@spaces.GPU(duration=120)
|
| 95 |
def load_model(self):
|
| 96 |
"""모델을 생성하고 GPU에 할당."""
|
| 97 |
model = stf_alternative.create_model(
|
|
|
|
| 103 |
)
|
| 104 |
return model
|
| 105 |
|
| 106 |
+
@spaces.GPU(duration=120)
|
| 107 |
def create_template(self):
|
| 108 |
"""템플릿 생성."""
|
| 109 |
template = stf_alternative.Template(
|