Update app.py

app.py

@@ -24,8 +24,8 @@ def upload_model(source_url, dst_repo, token,civitai_api_key, new_name, dst_repo
         if not dst_repo:
             raise gr.Error("You haven't provided a destination repository.")
             
-        #if not civitai_api_key:
-           #raise gr.Error("You haven't provided a Civitai Api Key.")
+        if not civitai_api_key:
+           raise gr.Error("You haven't provided a Civitai Api Key.")
             
         # Login to Hugging Face Hub
         login(token=token)
@@ -34,7 +34,8 @@ def upload_model(source_url, dst_repo, token,civitai_api_key, new_name, dst_repo
         download_dir = "/home/user/apps/downloads/" + str(int(time.time())) + str(random.getrandbits(8)) + "/"
         subprocess.check_call(["mkdir", "-p", download_dir])
         # Download the model using wget
-        subprocess.check_call(["wget", f"{source_url}", "-P", download_dir])
+        subprocess.check_call(["wget", f"{source_url}&token={civitai_api_key}", "-P", download_dir])
+
 
         # List files in the download directory
         files = os.listdir(download_dir)
@@ -76,9 +77,10 @@ def upload_model(source_url, dst_repo, token,civitai_api_key, new_name, dst_repo
 interface = gr.Interface(
     fn=upload_model,
     inputs=[
-        gr.Textbox(label="Source URL", placeholder="Source URL Civitai (e.g. https://civitai.com/api/download/models/486156)"),
+        gr.Textbox(label="Source URL", placeholder="Source URL Civitai (e.g. https://civitai.com/api/download/models/486156?type=Model&format=SafeTensor)"),
         gr.Textbox(label="Huggingface repository", placeholder="Destination repository HF (e.g. username/repo-name)"),
         gr.Textbox(label="Huggingface Token", placeholder="Write access token HF", type="password"),
+        gr.Textbox(label="Civitai Api Key", placeholder="Civitai Api Key", type="password"),
         gr.Textbox(label="New name", placeholder="New name for the model file (sdxl.safeTensor with file format)"),
         gr.Textbox(label="Huggingface repository Path", placeholder="Destination path within your repository (e.g. /models/Stable-diffusion/)"),
         gr.Dropdown(label="Repo Types", choices=REPO_TYPES, value="model"),