modified: main.py

main.py

@@ -110,7 +110,14 @@ def home_page():
             f.close()
             
             with st.spinner('Model Running....'):
-                col2.table(pred(st.session_state.image.name,radio,selbox,check))
+                res=pred(st.session_state.image.name,radio,selbox,check)
+            if check:
+                col2.write(res)
+            else :
+                col2.success(str(res))
+                
+                    
+