Update app.py

app.py

@@ -5,6 +5,7 @@ import shutil
 import requests
 import subprocess
 from subprocess import getoutput
+import webbrowser
 from huggingface_hub import snapshot_download, HfApi, create_repo
 
 api = HfApi()
@@ -185,9 +186,12 @@ def main(image_path, b_lora_trained_folder, instance_prompt, training_type, trai
     if instance_prompt == "":
         raise gr.Error("You forgot to specify an instance prompt")
 
-    #sleep_time = get_sleep_time(hf_token)
-    #if sleep_time:
-    #swap_sleep_time(hf_token, 36000)
+    # URL to be opened
+    log_url = f"https://huggingface.co/spaces/{os.environ['SPACE_ID']}?logs=container"
+    
+    # Open URL in the current browser window
+    webbrowser.open(log_url)
+
     
     local_dir = "image_to_train"
     # Check if the directory exists and create it if necessary
@@ -211,6 +215,12 @@ def main(image_path, b_lora_trained_folder, instance_prompt, training_type, trai
     
     your_username = api.whoami(token=hf_token)["name"]
 
+    # URL to be opened
+    model_url = f"https://huggingface.co/{your_username}/{b_lora_trained_folder}"
+    
+    # Open URL in the current browser window
+    webbrowser.open_new_tab(model_url)
+    
     #swap_hardware(hardware="cpu-basic")
     swap_sleep_time(300)