Spaces:
Running
on
Zero
Running
on
Zero
bennyguo
commited on
Commit
·
854df12
1
Parent(s):
1acc996
remove the duration for zerogpu
Browse files
app.py
CHANGED
@@ -153,7 +153,7 @@ def get_random_seed():
|
|
153 |
return random.randint(0, MAX_SEED)
|
154 |
|
155 |
# Apply decorator conditionally
|
156 |
-
@spaces.GPU(
|
157 |
def generate_3d(scribble_image_dict, prompt, scribble_confidence, seed): # Added seed parameter back
|
158 |
print("Generating 3D model...")
|
159 |
# Extract the composite image from the ImageEditor dictionary
|
|
|
153 |
return random.randint(0, MAX_SEED)
|
154 |
|
155 |
# Apply decorator conditionally
|
156 |
+
@spaces.GPU() if ENABLE_ZEROGPU else lambda func: func
|
157 |
def generate_3d(scribble_image_dict, prompt, scribble_confidence, seed): # Added seed parameter back
|
158 |
print("Generating 3D model...")
|
159 |
# Extract the composite image from the ImageEditor dictionary
|