Kohaku-Blueleaf
commited on
Commit
·
acdfa39
1
Parent(s):
2656dce
fixes
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ DEFAULT_FORMAT = """
|
|
57 |
|
58 |
def GPU(func, duration=None):
|
59 |
if in_spaces:
|
60 |
-
return spaces.GPU(func, duration)
|
61 |
else:
|
62 |
return func
|
63 |
|
|
|
57 |
|
58 |
def GPU(func, duration=None):
|
59 |
if in_spaces:
|
60 |
+
return spaces.GPU(func, duration=duration)
|
61 |
else:
|
62 |
return func
|
63 |
|