Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
02347e0
1
Parent(s):
05859a6
refactor(app): wrap create_app in spaces_app for HF Spaces GPU detection
Browse files
app.py
CHANGED
|
@@ -986,6 +986,17 @@ if SPACES_AVAILABLE:
|
|
| 986 |
if os.getenv("ENABLE_GPU", "false").lower() == "true":
|
| 987 |
create_app = spaces.GPU(create_app)
|
| 988 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 989 |
if __name__ == "__main__":
|
| 990 |
# Create and launch the application
|
| 991 |
try:
|
|
|
|
| 986 |
if os.getenv("ENABLE_GPU", "false").lower() == "true":
|
| 987 |
create_app = spaces.GPU(create_app)
|
| 988 |
|
| 989 |
+
# HuggingFace Spaces GPU function for deployment detection
|
| 990 |
+
if SPACES_AVAILABLE:
|
| 991 |
+
@spaces.GPU
|
| 992 |
+
def spaces_app():
|
| 993 |
+
"""HuggingFace Spaces GPU-enabled application function"""
|
| 994 |
+
return create_app()
|
| 995 |
+
else:
|
| 996 |
+
def spaces_app():
|
| 997 |
+
"""HuggingFace Spaces application function"""
|
| 998 |
+
return create_app()
|
| 999 |
+
|
| 1000 |
if __name__ == "__main__":
|
| 1001 |
# Create and launch the application
|
| 1002 |
try:
|