Update app.py

app.py

@@ -24,14 +24,15 @@ if file_name is not None:
     image = Image.open(file_name)
     col1.image(image, use_column_width=True)
 
-    label = st.text_input("What to look for in the image?", "Cats")
-
-    predictions = scan_image(image, label)
-    
-    col2.header("Probabilities") 
-    for key, value in predictions[1]:
-        col2.subheader(f"{ key }: { round(value * 100, 1)}%")
-
-    if predictions[1]:
-        st.header("The object is present in the given image")
-    else: st.header("The object is not found in the given image")
+    label = st.text_input("What to look for in the image?")
+
+    if st.button("Scan Image"):
+        predictions = scan_image(image, label)
+        
+        col2.header("Probabilities") 
+        for key in predictions[1].keys():
+            col2.subheader(f"{ key }: { round(predictions[1][key] * 100, 1)}%")
+
+        if predictions[0]:
+            st.header("The object is present in the given image")
+        else: st.header("The object is not found in the given image")