dwb2023 commited on
Commit
028d122
1 Parent(s): c7f454a

Update app.py

Browse files

remove gpu duration

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(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")
 
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")