SkalskiP commited on
Commit
bc6d5e6
1 Parent(s): 4ae7d54
Files changed (3) hide show
  1. README.md +2 -2
  2. app.py +2 -0
  3. requirements.txt +1 -0
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  title: Florence-2 for Videos
3
  emoji: 🎬
4
- colorFrom: green
5
- colorTo: pink
6
  sdk: gradio
7
  sdk_version: 4.37.2
8
  app_file: app.py
 
1
  ---
2
  title: Florence-2 for Videos
3
  emoji: 🎬
4
+ colorFrom: pink
5
+ colorTo: purple
6
  sdk: gradio
7
  sdk_version: 4.37.2
8
  app_file: app.py
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  from unittest.mock import patch
3
 
4
  import gradio as gr
@@ -68,6 +69,7 @@ def annotate_image(
68
  return output_image
69
 
70
 
 
71
  def process_video(
72
  input_video: str,
73
  progress=gr.Progress(track_tqdm=True)
 
1
  import os
2
+ import spaces
3
  from unittest.mock import patch
4
 
5
  import gradio as gr
 
69
  return output_image
70
 
71
 
72
+ @spaces.GPU
73
  def process_video(
74
  input_video: str,
75
  progress=gr.Progress(track_tqdm=True)
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  tqdm
2
  einops
 
3
  timm
4
  gradio
5
  transformers
 
1
  tqdm
2
  einops
3
+ spaces
4
  timm
5
  gradio
6
  transformers