Update app.py

app.py

@@ -21,14 +21,13 @@ ocean = Ocean(d)
 def wallet():
     
     address = connect("wallet")
-    st.write(address[0])
 
-    if address != "not":
+    if address[0] != "not":
 
         OCEAN_token = BToken(ocean.web3, ocean.OCEAN_address)
 
-        eth_balance = from_wei(ocean.web3.eth.get_balance(address))
-        ocean_balance = from_wei(OCEAN_token.balanceOf(address))
+        eth_balance = from_wei(ocean.web3.eth.get_balance(address[0]))
+        ocean_balance = from_wei(OCEAN_token.balanceOf(address[0]))
 
         st.write(f'Address: {address}')
         st.write(f'ETH Balance: {eth_balance}')