Update app.py

app.py

@@ -258,31 +258,25 @@ def start_training(
     ## Your model is available at: <a href='https://huggingface.co/{username}/{slugged_lora_name}'>{username}/{slugged_lora_name}</a>"""
 
 def swap_visibilty(profile: Union[gr.OAuthProfile, None]):
-    if is_spaces:
-        if profile is None:
-            return gr.update(elem_classes=["main_ui_logged_out"])
-        else:
-            return gr.update(elem_classes=["main_ui_logged_in"])
-    else:
-        return gr.update(elem_classes=["main_ui_logged_in"])
-
+    # HF_TOKEN이 있으므로 항상 로그인된 상태로 처리
+    return gr.update(elem_classes=["main_ui_logged_in"])
+    
 def update_pricing(steps, oauth_token: Union[gr.OAuthToken, None]):
-    if(oauth_token and is_spaces):
-        user = whoami(oauth_token.token)
+    # HF_TOKEN을 사용하여 직접 확인
+    try:
+        user = whoami()
         seconds_per_iteration = 7.54
         total_seconds = (steps * seconds_per_iteration) + 240
         cost_per_second = 0.80/60/60
         cost = round(cost_per_second * total_seconds, 2)
         cost_preview = f'''To train this LoRA, a paid L4 GPU will be hooked under the hood during training and then removed once finished.
         ### Estimated to cost <b>< US$ {str(cost)}</b> for {round(int(total_seconds)/60, 2)} minutes with your current train settings <small>({int(steps)} iterations at {seconds_per_iteration}s/it)</small>'''
-        if(user["canPay"]):
-            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), pay_disclaimer+cost_preview, gr.update(visible=True), gr.update(visible=False)
-    else:
+        return gr.update(visible=True), cost_preview, gr.update(visible=False), gr.update(visible=True)
+    except:
         return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
 
+
+
 def swap_base_model(model):
     return gr.update(visible=True) if model == "[dev] (high quality model, non-commercial license)" else gr.update(visible=False)    
 
@@ -329,14 +323,14 @@ theme = gr.themes.Monochrome(
     text_size=gr.themes.Size(lg="18px", md="15px", sm="13px", xl="22px", xs="12px", xxl="24px", xxs="9px"),
     font=[gr.themes.GoogleFont("Source Sans Pro"), "ui-sans-serif", "system-ui", "sans-serif"],
 )
+
+
 css = """
 h1{font-size: 2em}
 h3{margin-top: 0}
 #component-1{text-align:center}
-.main_ui_logged_out{opacity: 0.3; pointer-events: none}
 .tabitem{border: 0px}
 .group_padding{padding: .55em}
-#space_model .wrap > label:last-child{opacity: 0.3; pointer-events:none}
 """
 
 # LoginButton 부분 제거