Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
1e1a102
1
Parent(s):
cc0a32f
fix bug
Browse files
utils/__pycache__/interface.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/interface.cpython-310.pyc and b/utils/__pycache__/interface.cpython-310.pyc differ
|
|
|
utils/interface.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import io
|
| 3 |
import os
|
|
@@ -35,6 +36,7 @@ class Inference:
|
|
| 35 |
model_path, trust_remote_code=True, device_map="auto"
|
| 36 |
).eval()
|
| 37 |
|
|
|
|
| 38 |
def forward(
|
| 39 |
self,
|
| 40 |
task: str,
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
import io
|
| 4 |
import os
|
|
|
|
| 36 |
model_path, trust_remote_code=True, device_map="auto"
|
| 37 |
).eval()
|
| 38 |
|
| 39 |
+
@spaces.GPU(duration = 120)
|
| 40 |
def forward(
|
| 41 |
self,
|
| 42 |
task: str,
|