Segizu commited on
Commit
fdd0cdf
Β·
1 Parent(s): 22bb6d2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ import time
14
  import shutil
15
  import tarfile
16
  import tensorflow as tf
17
- from spaces import GPU, ZeroGPU
18
 
19
 
20
  cuda_available = tf.config.list_physical_devices('GPU')
@@ -70,7 +70,7 @@ dataset = load_dataset(
70
  column_names=["image"],
71
  header=0
72
  )
73
- @ZeroGPU
74
  def build_database():
75
  print(f"πŸ“Š Uso actual de almacenamiento temporal INICIO: {get_folder_size('.'):.2f} GB")
76
  print("πŸ”„ Generando embeddings...")
 
14
  import shutil
15
  import tarfile
16
  import tensorflow as tf
17
+ from spaces import GPU
18
 
19
 
20
  cuda_available = tf.config.list_physical_devices('GPU')
 
70
  column_names=["image"],
71
  header=0
72
  )
73
+ @GPU
74
  def build_database():
75
  print(f"πŸ“Š Uso actual de almacenamiento temporal INICIO: {get_folder_size('.'):.2f} GB")
76
  print("πŸ”„ Generando embeddings...")