Update app.py

app.py

@@ -22,11 +22,11 @@ def compute(image):
         st.image(image, use_column_width=True)
 
     with st.container():
-        st.write("#### Different classification outputs at different threshold values:")
+        st.write("#### Classification Outputs:")
         col1, col2, col6 = st.columns(3)
-        col1.metric(predictions[0]['label'], str(round(predictions[0]['score']+100, 1))+"%")
-        col2.metric(predictions[1]['label'], str(round(predictions[1]['score']+100, 1))+"%")
-        col6.metric(predictions[2]['label'], str(round(predictions[2]['score']+100, 1))+"%")
+        col1.metric(predictions[0]['label'], str(round(predictions[0]['score']*100, 1))+"%")
+        col2.metric(predictions[1]['label'], str(round(predictions[1]['score']*100, 1))+"%")
+        col6.metric(predictions[2]['label'], str(round(predictions[2]['score']*100, 1))+"%")
     return None
 
 # INIT
@@ -38,22 +38,18 @@ st.write("# Fire in Forest Environments")
 st.write("""Wildfires or forest fires are unpredictable catastrophic and destructive events that affect rural areas. 
 The impact of these events affects both vegetation and wildlife. 
 
-This application showcases the "vit-fire-detection" model, a version of google vit-base-patch16-224-in21k vision transformer fine-tuned for smoke and fire detection.  
-In particular, we can imagine a setup in which webcams, drones, or other recording devices take pictures of a wild environment every t seconds or minutes. The proposed system is then able to classify the current situation as normal, smoke, or fire. 
-
-The system is not real-time since wildfires are events that evolve more slowly than others (e.g., traffic accidents).
+This application showcases the "vit-fire-detection" model, a version of google vit-base-patch16-224-in21k vision transformer fine-tuned for smoke and fire detection. In particular, we can imagine a setup in which webcams, drones, or other recording devices take pictures of a wild environment every t seconds or minutes. The proposed system is then able to classify the current situation as normal, smoke, or fire. 
 """)
 #st.image("./demo.jpg", use_column_width=True)
 st.write("#### Upload an image to see the classifier in action")
-
-demo_img = Image.open("./demo.jpg")
-compute(demo_img)
-
 # INPUT IMAGE
 file_name = st.file_uploader("")
 if file_name is not None:
     image = Image.open(file_name)
     compute(image)
 
+demo_img = Image.open("./demo.jpg")
+compute(demo_img)
+
 # SIDEBAR
 #st.sidebar.write("""""")