Update app.py

app.py

@@ -56,10 +56,8 @@ trainer = Trainer(HF_TOKEN)
 with gr.Blocks(css='style.css') as demo:
     if os.getenv('IS_SHARED_UI'):
         show_warning(SHARED_UI_WARNING)
-    if not torch.cuda.is_available():
+    elif not torch.cuda.is_available():
         show_warning(CUDA_NOT_AVAILABLE_WARNING)
-    if not HF_TOKEN:
-        show_warning(HF_TOKEN_NOT_SPECIFIED_WARNING)
 
     gr.Markdown(TITLE)
     with gr.Tabs():
@@ -72,5 +70,8 @@ with gr.Blocks(css='style.css') as demo:
             - You can use this tab to upload models later if you choose not to upload models in training time or if upload in training time failed.
             ''')
             create_upload_demo(HF_TOKEN)
+    
+    if not HF_TOKEN:
+        show_warning(HF_TOKEN_NOT_SPECIFIED_WARNING)
 
 demo.queue(max_size=1).launch(share=False)