Update app.py

app.py

@@ -264,14 +264,8 @@ def start_training(
     return f"""# Training completed successfully! 
     ## 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]):
-    # HF_TOKEN이 있으므로 항상 로그인된 상태로 처리
-    return gr.update(elem_classes=["main_ui_logged_in"])
-    
-def update_pricing(steps, oauth_token: Union[gr.OAuthToken, None]):
-    # HF_TOKEN을 사용하여 직접 확인
+def update_pricing(steps):
     try:
-        user = whoami()
         seconds_per_iteration = 7.54
         total_seconds = (steps * seconds_per_iteration) + 240
         cost_per_second = 0.80/60/60
@@ -281,7 +275,7 @@ 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)
     except:
         return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
-
+    
 
 
 def swap_base_model(model):
@@ -346,8 +340,10 @@ with gr.Blocks(theme=theme, css=css) as demo:
         """# 🆔 Gini LoRA 학습
 ### 이미지들(최대 150장 아니)을 업로드하세요. """
     )
-    with gr.Tab("Train"):  # "Train on Spaces" 대신 단순히 "Train"으로 변경
-        with gr.Column() as main_ui:
+    
+    with gr.Tab("Train"):  # 탭 이름 변경
+        with gr.Column():  # main_ui 대신 직접 Column 사용
+            with gr.Group():
 
             with gr.Group():
                 with gr.Row():
@@ -506,38 +502,33 @@ with gr.Blocks(theme=theme, css=css) as demo:
     )
     
     gr.on(
-        triggers=[steps.change, payment_update.click],
-        fn=update_pricing,
-        inputs=[steps],
-        outputs=[cost_preview, cost_preview_info, payment_update, start]
-    )
-
-    which_model.change(
-        fn=swap_base_model,
-        inputs=which_model,
-        outputs=model_warning
-    )
-    start.click(fn=create_dataset, inputs=[images] + caption_list, outputs=dataset_folder).then(
-        fn=start_training,
-        inputs=[
-            lora_name,
-            concept_sentence,
-            which_model,
-            steps,
-            lr,
-            rank,
-            dataset_folder,
-            sample_1,
-            sample_2,
-            sample_3,
-            use_more_advanced_options,
-            more_advanced_options
+            triggers=[steps.change],
+            fn=update_pricing,
+            inputs=[steps],
+            outputs=[cost_preview, cost_preview_info, payment_update, start]
+        )
+    
+        start.click(fn=create_dataset, inputs=[images] + caption_list, outputs=dataset_folder).then(
+            fn=start_training,
+            inputs=[
+                lora_name,
+                concept_sentence,
+                which_model,
+                    steps,
+                lr,
+                rank,
+                dataset_folder,
+                sample_1,
+                sample_2,
+                sample_3,
+                use_more_advanced_options,
+                    more_advanced_options
         ],
-        outputs=progress_area,
-    )
+            outputs=progress_area,
+        )
+
+        do_captioning.click(fn=run_captioning, inputs=[images, concept_sentence] + caption_list, outputs=caption_list)
 
-    do_captioning.click(fn=run_captioning, inputs=[images, concept_sentence] + caption_list, outputs=caption_list)
-    demo.load(fn=swap_visibilty, outputs=main_ui)
 
 if __name__ == "__main__":
     demo.launch(