max7777 commited on
Commit
a4937ed
1 Parent(s): f68ae5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -46,8 +46,7 @@ else:
46
  image = Image.open(file)
47
  st.image(image, use_column_width=True)
48
  predictions = upload_predict(image, model)
49
- image_class = str(predictions[0][0][1])
50
- score=np.round(predictions[0][0][2])
51
  st.write("The image is classified as",image_class)
52
- st.write("The similarity score is approximately",score)
53
- print("The image is classified as ",image_class, "with a similarity score of",score)
 
46
  image = Image.open(file)
47
  st.image(image, use_column_width=True)
48
  predictions = upload_predict(image, model)
49
+ image_class = str(predictions)
50
+
51
  st.write("The image is classified as",image_class)
52
+