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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -34,14 +34,7 @@ def classify_video(video):
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]}"
45
  Description="An MobileNET model trained to classify Fire and Smoke through Videos"
46
  Article="Created at jupyter NoteBook with GPU NVIDIA_GeForce_MX350"
47
  gr.Interface(
 
34
  Smoke.append(i)
35
  else:
36
  Default.append(i)
37
+ return predictions
 
 
 
 
 
 
 
38
  Description="An MobileNET model trained to classify Fire and Smoke through Videos"
39
  Article="Created at jupyter NoteBook with GPU NVIDIA_GeForce_MX350"
40
  gr.Interface(