rajkumarrawal commited on
Commit
02347e0
·
1 Parent(s): 05859a6

refactor(app): wrap create_app in spaces_app for HF Spaces GPU detection

Browse files
Files changed (1) hide show
  1. app.py +11 -0
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: