Update app.py
Browse files
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}")
|