Upload app.py

app.py

@@ -23,37 +23,51 @@ import huggingface_hub
 huggingface_hub.login(os.getenv('HF_TOKEN_FLUX'))
 
 
-def download_from_github(repo, filepath, destination, token):
+def download_release_asset(repo, release_tag, asset_name, destination, token):
     """
-    Download a file from a private GitHub repository.
-    
+    Download a file from a GitHub Release.
+
     Args:
         repo (str): The GitHub repository in the format "owner/repo".
-        filepath (str): The path to the file in the repository.
+        release_tag (str): The release tag (e.g., "v1.0.0").
+        asset_name (str): The name of the file in the release.
         destination (str): The local file path to save the downloaded file.
         token (str): The GitHub Personal Access Token.
-    
+
     Returns:
         None
     """
-    url = f"https://api.github.com/repos/{repo}/contents/{filepath}"
+    # Fetch release assets
+    url = f"https://api.github.com/repos/{repo}/releases/tags/{release_tag}"
     headers = {"Authorization": f"token {token}"}
     response = requests.get(url, headers=headers)
-    
+
     if response.status_code == 200:
-        file_content = response.json().get('content')
-        if file_content:
+        release_data = response.json()
+        # Find the asset by name
+        asset = next((a for a in release_data['assets'] if a['name'] == asset_name), None)
+        if not asset:
+            raise Exception(f"Asset {asset_name} not found in release {release_tag}.")
+
+        # Download the asset
+        download_url = asset['browser_download_url']
+        download_response = requests.get(download_url, headers=headers, stream=True)
+
+        if download_response.status_code == 200:
             os.makedirs(os.path.dirname(destination), exist_ok=True)
             with open(destination, 'wb') as f:
-                f.write(base64.b64decode(file_content))
-            print(f"Downloaded {filepath} to {destination}")
+                for chunk in download_response.iter_content(chunk_size=1024):
+                    f.write(chunk)
+            print(f"Downloaded {asset_name} to {destination}")
         else:
-            raise Exception(f"File content not found for {filepath} in {repo}")
+            raise Exception(f"Failed to download {asset_name}. Status code: {download_response.status_code}")
     else:
-        raise Exception(f"Failed to download {filepath} from {repo}. Status code: {response.status_code}")
+        raise Exception(f"Failed to fetch release {release_tag}. Status code: {response.status_code}")
+
 
 # Example: Define GitHub repo and file paths
 repo_name = "arj7192/NativDemoLocal"
+release_tag = "v1"
 github_files = {
     "speech_bubble_model.pt": "speech_bubble_model.pt",
     "craft_mlt_25k.pth": "craft_mlt_25k.pth",
@@ -68,7 +82,7 @@ token = os.getenv("HF_GITHUB_TOKEN")
 # Download each file
 for filename, repo_path in github_files.items():
     destination_path = f"{filename}"
-    download_from_github(repo_name, repo_path, destination_path, token)
+    download_release_asset(repo_name, release_tag, filename, destination_path, token)
 
 bubble_detection_model = YOLO("speech_bubble_model.pt")