sayakpaul HF staff commited on
Commit
dcecbc8
1 Parent(s): d1ed60e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ from PIL import Image
11
  from create_maxim_model import Model
12
  from maxim.configs import MAXIM_CONFIGS
13
 
 
14
  _MODEL = from_pretrained_keras("sayakpaul/S-2_enhancement_lol")
15
 
16
 
 
11
  from create_maxim_model import Model
12
  from maxim.configs import MAXIM_CONFIGS
13
 
14
+ print(f"Number of GPUs detected by TF: {tf.config.list_physical_devices('GPU')}")
15
  _MODEL = from_pretrained_keras("sayakpaul/S-2_enhancement_lol")
16
 
17