Initial Release

app.py

@@ -31,8 +31,8 @@ def processImage():
         cv2.imwrite('data/result.jpg', result_frame)
         # display the image and class
         img_ = Image.open(f"data/result.jpg")
-        result_class = labels.split()[0]
-        confidence = float(labels.split()[1])
+        result_class = str(labels).split()[0]
+        confidence = float(str(labels).split()[1])
         
         st.subheader(f"Result {result_class} with confidence {confidence * 100 :.2f}%")
         st.image(img_)