Spaces:
Sleeping
Sleeping
functionNormally commited on
Commit ·
8e71d97
1
Parent(s): ddfedb8
Ajouter @spaces.GPU à extract_features_callback (ZeroGPU)
Browse files
app.py
CHANGED
|
@@ -39,6 +39,7 @@ def refresh_gallery_callback(split_name, class_name, max_images):
|
|
| 39 |
# Tab 2 — ML classique
|
| 40 |
# ---------------------------------------------------------------------------
|
| 41 |
|
|
|
|
| 42 |
def extract_features_callback():
|
| 43 |
try:
|
| 44 |
_, class_names, counts = extract_all_features()
|
|
|
|
| 39 |
# Tab 2 — ML classique
|
| 40 |
# ---------------------------------------------------------------------------
|
| 41 |
|
| 42 |
+
@spaces.GPU(duration=60)
|
| 43 |
def extract_features_callback():
|
| 44 |
try:
|
| 45 |
_, class_names, counts = extract_all_features()
|