Add application file

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from tensorflow import keras
+import tensorflow as tf
+
+model = tf.keras.models.load_model("thebestteamever/fire_detection_project/model.h5")
+
+#return a dictionary of labels and probabilities
+def fire_or_non_fire(img):
+    img = img.reshape((-1 ,105, 165, 3))
+    #img = np.array(img)
+    prediction = model.predict(img).tolist()[0]
+    labels = ["Fire", "Non Fire"]
+    return {labels[i]: prediction[i] for i in range(2)}
+
+im = gr.inputs.Image(shape=(105, 165), source="upload")
+iface = gr.Interface(
+    fn = fire_or_non_fire, 
+    inputs = im, 
+    outputs = gr.outputs.Label(),
+)
+
+iface.launch(share=True, debug=False)