Update app.py

app.py

@@ -104,9 +104,8 @@ img = st.file_uploader('', type=['jpeg', 'jpg', 'png'])
 
 # display image
 if img is not None:
-    image = Image.open(img).convert('RGB')
-    st.image(img, use_column_width=True)
+    image = Image.open(img)
+    st.image(image, use_column_width=True)
 
 
-    # write classification
-    st.write("## {}".format(FINAL(img)))
+st.write(FINAL(image))