Martin Tomov commited on
Commit
04092d2
Β·
verified Β·
1 Parent(s): d969435

extract_and_paste_insect and create_yellow_background_with_insects fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ class DetectionResult:
30
  score: float
31
  label: str
32
  box: BoundingBox
33
- mask: Optional[np.array] = None
34
 
35
  @classmethod
36
  def from_dict(cls, detection_dict: Dict) -> 'DetectionResult':
@@ -169,4 +169,4 @@ gr.Interface(
169
  inputs=gr.Image(type="pil"),
170
  outputs=[gr.Image(type="numpy"), gr.Image(type="numpy")],
171
  title="🐞 Insect Detection and Masking"
172
- ).launch()
 
30
  score: float
31
  label: str
32
  box: BoundingBox
33
+ mask: Optional[np.ndarray] = None
34
 
35
  @classmethod
36
  def from_dict(cls, detection_dict: Dict) -> 'DetectionResult':
 
169
  inputs=gr.Image(type="pil"),
170
  outputs=[gr.Image(type="numpy"), gr.Image(type="numpy")],
171
  title="🐞 Insect Detection and Masking"
172
+ ).launch()