Update app.py

#2
by reach-vb HF staff - opened
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,8 @@ import requests
4
  import json
5
  from huggingface_hub import HfApi, create_repo
6
 
 
 
7
  def download_file(digest, image):
8
  url = f"https://registry.ollama.ai/v2/library/{image}/blobs/{digest}"
9
  file_name = f"blobs/{digest}"
@@ -29,7 +31,7 @@ def fetch_manifest(image, tag):
29
  return None
30
 
31
  def upload_to_huggingface(repo_id, folder_path):
32
- api = HfApi()
33
  repo_path = create_repo(repo_id, "model", exists_ok=True)
34
  print(f"Repo created {repo_path}")
35
  try:
 
4
  import json
5
  from huggingface_hub import HfApi, create_repo
6
 
7
+ HF_TOKEN = os.environ.get("HF_TOKEN")
8
+
9
  def download_file(digest, image):
10
  url = f"https://registry.ollama.ai/v2/library/{image}/blobs/{digest}"
11
  file_name = f"blobs/{digest}"
 
31
  return None
32
 
33
  def upload_to_huggingface(repo_id, folder_path):
34
+ api = HfApi(token=HF_TOKEN)
35
  repo_path = create_repo(repo_id, "model", exists_ok=True)
36
  print(f"Repo created {repo_path}")
37
  try: