Update app.py

app.py

@@ -34,7 +34,14 @@ def classify_video(video):
             Smoke.append(i)
         else:
             Default.append(i)
-    return predictions
+    if len(Fire)>10 and len(Smoke)>8:
+        return f"Spotted {class_names[1]} and {class_names[2]}"
+    elif len(Fire)>10:
+        return f"Spotted {class_names[1]}"
+    elif len(Smoke)>8:
+        return f"Spotted {class_names[2]}"
+    else:
+        return f"Spotted {class_names[0]}"
 Description="An MobileNET model trained to classify Fire and Smoke through Videos"
 Article="Created at jupyter NoteBook with GPU NVIDIA_GeForce_MX350"
 gr.Interface(