bhardwaj08sarthak commited on
Commit
fdbf928
·
verified ·
1 Parent(s): 896413c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -419,7 +419,7 @@ with gr.Blocks() as demo:
419
 
420
  if __name__ == "__main__" or os.getenv("SYSTEM") == "spaces":
421
  try:
422
- load_model() # triggers GPU allocation during startup
423
  except Exception as e:
424
  # don't crash the app if warmup fails; logs will show details
425
  print("Warmup failed:", e)
 
419
 
420
  if __name__ == "__main__" or os.getenv("SYSTEM") == "spaces":
421
  try:
422
+ get_local_model_gpu() # triggers GPU allocation during startup
423
  except Exception as e:
424
  # don't crash the app if warmup fails; logs will show details
425
  print("Warmup failed:", e)