Spaces:
Runtime error
Runtime error
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
|