Update app.py

app.py

@@ -69,13 +69,13 @@ def download_file(url, filename, folder="."):
     except requests.exceptions.RequestException as e:
         print(f"Error downloading file: {e}")
 
-def process_url(url, download_files=True, folder="."):
+def process_url(url, do_download=True, folder="."):
     json_data = get_json_data(url)
     if json_data:
         if check_nsfw(json_data):
             info = extract_info(json_data)
             if info:
-                if(download_files):
+                if(do_download):
                     downloaded_files = download_files(info, folder)
                 else:
                     downloaded_files = []
@@ -127,7 +127,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
     info, downloaded_files = process_url(url, folder)
     create_readme(info, downloaded_files, folder)
     try:
-        api = HfApi(token=hf_token)
+        api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])
         username = api.whoami()["name"]
         slug_name = slugify(info["name"])
         repo_id = f"{username}/{slug_name}"
@@ -137,9 +137,12 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
             repo_id=repo_id,
             repo_type="model"
         )
+        transfer_repos = gr.load("multimodalart/transfer_repos", api_key=os.environ["HUGGING_FACE_HUB_TOKEN"])
+        transfer_repos(repo_id, f"{profile.preferred_username}/{slug_name}")
     except:
         raise gr.Error("something went wrong")
-    return "Model uploaded!"
+    return f'''### Model uploaded!
+    Access it here [{profile.preferred_username}/{slug_name}](https://huggingface.co/{profile.preferred_username}/{slug_name}) '''
 
 
 def get_creator(username):
@@ -245,7 +248,7 @@ with gr.Blocks(css=css) as demo:
         instructions = gr.HTML("")
         try_again_button = gr.Button("I have added my HF profile to my account", visible=False)
         submit_button_civit = gr.Button("Upload model to Hugging Face", interactive=False)
-        output = gr.Textbox(label="Output progress", visible=False)
+        output = gr.Markdown(label="Output progress", visible=False)
    
     demo.load(fn=swap_fill, outputs=[disabled_area, enabled_area])
     submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit, try_again_button])