Update app.py

app.py

@@ -22,17 +22,15 @@ ocean = Ocean(d)
 def wallet():
     
     lower_case_address = connect("wallet")
-    st.write(lower_case_address[0][0:2])
     address = ''
     if len(lower_case_address[0]) > 3:
         address = Web3.toChecksumAddress(lower_case_address[0])
-    st.write(len(address))
     if len(address) > 3:
 
         OCEAN_token = BToken(ocean.web3, ocean.OCEAN_address)
 
-        eth_balance = from_wei(ocean.web3.eth.get_balance(address[0]))
-        ocean_balance = from_wei(OCEAN_token.balanceOf(address[0]))
+        eth_balance = from_wei(ocean.web3.eth.get_balance(address))
+        ocean_balance = from_wei(OCEAN_token.balanceOf(address))
 
         st.write(f'Address: {address}')
         st.write(f'ETH Balance: {eth_balance}')