Update app.py

app.py

@@ -323,7 +323,10 @@ if uploaded_image and query:
         for class_name, confidence in predictions:
             if int(confidence) > 0.05:
                 fpredictions += f"{class_name}: {confidence:.2f}%,"
-            st.write(f"{class_name}: {confidence:.2f}%")
+            if int(confidence) > 5:
+                class_name = class_name.replace("_", " ")
+                class_name = class_name.title()
+                st.write(f"{class_name}: {confidence:.2f}%")
         print(fpredictions)
 
     # call openai to pick the best classification result based on query
@@ -372,7 +375,10 @@ elif uploaded_image is not None:
         for class_name, confidence in predictions:
             if int(confidence) > 0.05:
                 fpredictions += f"{class_name}: {confidence:.2f}%,"
-            st.write(f"{class_name}: {confidence:.2f}%")
+            if int(confidence) > 5:
+                class_name = class_name.replace("_", " ")
+                class_name = class_name.title()
+                st.write(f"{class_name}: {confidence:.2f}%")
         print(fpredictions)
 
 elif query: