reab5555 commited on
Commit
813b964
·
verified ·
1 Parent(s): a6ffc73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,6 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
27
  import tensorflow as tf
28
  tf.get_logger().setLevel('ERROR')
29
 
30
- @spaces.GPU(duration=500)
31
 
32
  # Initialize models and other global variables
33
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -356,6 +355,7 @@ def get_random_face_sample(organized_faces_folder, largest_cluster, output_folde
356
  return output_path
357
  return None
358
 
 
359
  def process_video(video_path, num_anomalies, num_components, desired_fps, batch_size, progress=gr.Progress()):
360
  print(f"Video path: {video_path}")
361
  print(f"Num anomalies: {num_anomalies}")
 
27
  import tensorflow as tf
28
  tf.get_logger().setLevel('ERROR')
29
 
 
30
 
31
  # Initialize models and other global variables
32
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
355
  return output_path
356
  return None
357
 
358
+ @spaces.GPU(duration=500)
359
  def process_video(video_path, num_anomalies, num_components, desired_fps, batch_size, progress=gr.Progress()):
360
  print(f"Video path: {video_path}")
361
  print(f"Num anomalies: {num_anomalies}")