Anshu13 commited on
Commit
d99e696
·
verified ·
1 Parent(s): 796503a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -34,11 +34,11 @@ def classify_video(video):
34
  Smoke.append(i)
35
  else:
36
  Default.append(i)
37
- if Fire!=[] and Smoke!=[]:
38
  return f"Spotted {class_names[1]} and {class_names[2]}"
39
- elif Fire!=[]:
40
  return f"Spotted {class_names[1]}"
41
- elif Smoke!=[]:
42
  return f"Spotted {class_names[2]}"
43
  else:
44
  return f"Spotted {class_names[0]}"
 
34
  Smoke.append(i)
35
  else:
36
  Default.append(i)
37
+ if len(Fire)>5 and len(Smoke)>5:
38
  return f"Spotted {class_names[1]} and {class_names[2]}"
39
+ elif len(Fire)>5:
40
  return f"Spotted {class_names[1]}"
41
+ elif len(Smoke)>5:
42
  return f"Spotted {class_names[2]}"
43
  else:
44
  return f"Spotted {class_names[0]}"