dwb2023 commited on
Commit
bd4f934
·
verified ·
1 Parent(s): 043fb30

Update app.py

Browse files

@spaces.GPU(duration=120)

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ subprocess.run(
15
  hf_token = os.getenv("HF_TOKEN")
16
  login(token=hf_token, add_to_git_credential=True)
17
 
18
- @spaces.GPU
19
  def get_model_summary(model_name):
20
  # Check if CUDA is available and set the device accordingly
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
15
  hf_token = os.getenv("HF_TOKEN")
16
  login(token=hf_token, add_to_git_credential=True)
17
 
18
+ @spaces.GPU(duration=120)
19
  def get_model_summary(model_name):
20
  # Check if CUDA is available and set the device accordingly
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")