Segizu commited on
Commit
22bb6d2
Β·
1 Parent(s): 5bc3fb1
Files changed (1) hide show
  1. app.py +2 -3
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
18
 
19
 
20
  cuda_available = tf.config.list_physical_devices('GPU')
@@ -70,8 +70,7 @@ dataset = load_dataset(
70
  column_names=["image"],
71
  header=0
72
  )
73
-
74
-
75
  def build_database():
76
  print(f"πŸ“Š Uso actual de almacenamiento temporal INICIO: {get_folder_size('.'):.2f} GB")
77
  print("πŸ”„ Generando embeddings...")
 
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
  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...")