Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
from loadimg import load_img
|
3 |
-
import spaces
|
4 |
from transformers import AutoModelForImageSegmentation
|
5 |
import torch
|
6 |
from torchvision import transforms
|
@@ -49,7 +49,7 @@ def process_frame(frame, bg_type, bg, fast_mode, bg_frame_index, background_fram
|
|
49 |
print(f"Error processing frame: {e}")
|
50 |
return frame, bg_frame_index
|
51 |
|
52 |
-
|
53 |
def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=0, video_handling="slow_down", fast_mode=True, max_workers=10):
|
54 |
try:
|
55 |
start_time = time.time() # Start the timer
|
@@ -188,4 +188,4 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
|
|
188 |
)
|
189 |
|
190 |
if __name__ == "__main__":
|
191 |
-
demo.launch(show_error=True)
|
|
|
1 |
import gradio as gr
|
2 |
from loadimg import load_img
|
3 |
+
#import spaces
|
4 |
from transformers import AutoModelForImageSegmentation
|
5 |
import torch
|
6 |
from torchvision import transforms
|
|
|
49 |
print(f"Error processing frame: {e}")
|
50 |
return frame, bg_frame_index
|
51 |
|
52 |
+
#@spaces.GPU
|
53 |
def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=0, video_handling="slow_down", fast_mode=True, max_workers=10):
|
54 |
try:
|
55 |
start_time = time.time() # Start the timer
|
|
|
188 |
)
|
189 |
|
190 |
if __name__ == "__main__":
|
191 |
+
demo.launch(show_error=True, share = True)
|