Update app.py

app.py

@@ -1,9 +1,10 @@
 import os
 import requests
 import gradio as gr
-from huggingface_hub import Repository
+from huggingface_hub import upload_file, HfFolder
 
 REPO_ID = "AthuKawaleLogituit/Faceswap"
+TOKEN = os.environ.get("HF_TOKEN")
 LOCAL_DIR = "models"
 
 urls = [
@@ -24,22 +25,36 @@ urls = [
     "https://huggingface.co/countfloyd/deepfake/resolve/main/lsdir_x4.onnx"
 ]
 
+def download_file(url, filepath):
+    print(f"Downloading {url}")
+    with requests.get(url, stream=True, allow_redirects=True) as r:
+        r.raise_for_status()
+        with open(filepath, 'wb') as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                if chunk:
+                    f.write(chunk)
+
 def download_and_upload():
     os.makedirs(LOCAL_DIR, exist_ok=True)
+    uploaded_files = []
 
     for url in urls:
-        filename = os.path.join(LOCAL_DIR, os.path.basename(url))
-        if not os.path.exists(filename):
-            r = requests.get(url, allow_redirects=True)
-            with open(filename, 'wb') as f:
-                f.write(r.content)
-
-    token = os.environ.get("HF_TOKEN")
-    repo = Repository(local_dir=LOCAL_DIR, clone_from=REPO_ID, use_auth_token=token)    
-    repo.git_add()
-    repo.git_commit("Upload model files from Gradio Space")
-    repo.git_push()
-
-    return "✅ Models downloaded and pushed to Faceswap repo."
+        filename = os.path.basename(url)
+        local_path = os.path.join(LOCAL_DIR, filename)
+
+        if not os.path.exists(local_path):
+            download_file(url, local_path)
+
+        # Upload to repo
+        upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=filename,
+            repo_id=REPO_ID,
+            repo_type="model",
+            token=TOKEN
+        )
+        uploaded_files.append(filename)
+
+    return f"✅ Uploaded files: {uploaded_files}"
 
 gr.Interface(fn=download_and_upload, inputs=[], outputs="text", title="Upload Faceswap Models").launch()