fixes

app.py

@@ -2,6 +2,7 @@ import os
 import random
 import json
 from pathlib import Path
+from functools import partial
 
 if os.environ.get("IN_SPACES", None) is not None:
     in_spaces = True
@@ -60,7 +61,9 @@ DEFAULT_FORMAT = """
 """.strip()
 
 
-def GPU(func, duration=None):
+def GPU(func=None, duration=None):
+    if func is None:
+        return partial(GPU, duration=duration)
     if in_spaces:
         if duration:
             return spaces.GPU(func, duration=duration)