SkalskiP commited on
Commit
fb9c81b
·
1 Parent(s): 34cb512

GPU Zero test

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from typing import Tuple
2
 
 
3
  import gradio as gr
4
  import numpy as np
5
  import supervision as sv
@@ -52,6 +53,7 @@ def detect_and_annotate(
52
  return annotated_image
53
 
54
 
 
55
  def process_image(
56
  input_image: np.ndarray,
57
  confidence_threshold: float,
 
1
  from typing import Tuple
2
 
3
+ import spaces
4
  import gradio as gr
5
  import numpy as np
6
  import supervision as sv
 
53
  return annotated_image
54
 
55
 
56
+ @spaces.GPU(duration=200)
57
  def process_image(
58
  input_image: np.ndarray,
59
  confidence_threshold: float,