Update app.py

app.py

@@ -293,9 +293,9 @@ def update_pricing(steps, oauth_token: Union[gr.OAuthToken, None]):
             return gr.update(visible=True), cost_preview, gr.update(visible=False), gr.update(visible=True)
         else:
             pay_disclaimer = f'''## ⚠️ {user["name"]}, your account doesn't have a payment method. Set one up <a href='https://huggingface.co/settings/billing/payment' target='_blank'>here</a> and come back here to train your LoRA<br><br>'''
-            return gr.update(visible=True),
-
-
+            return gr.update(visible=True), pay_disclaimer+cost_preview, gr.update(visible=True), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
 
 return gr.update(visible=True), pay_disclaimer+cost_preview, gr.update(visible=True), gr.update(visible=False)
     else: