pdufour commited on
Commit
ca68348
1 Parent(s): 5b280f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -8
app.py CHANGED
@@ -24,24 +24,16 @@ ARCHIVE_PATH = f"./transformers_{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
24
  HF_BASE_URL = "https://huggingface.co"
25
 
26
  if not os.path.exists(TRANSFORMERS_REPOSITORY_PATH):
27
- # Download the .tar.gz file
28
- print(f"Downloading the repository from {TRANSFORMERS_REPOSITORY_URL}...")
29
  urllib.request.urlretrieve(TRANSFORMERS_REPOSITORY_URL, ARCHIVE_PATH)
30
 
31
- # Create a temporary directory for extraction
32
  with tempfile.TemporaryDirectory() as tmp_dir:
33
- # Extract the .tar.gz file to temp directory
34
- print(f"Extracting the archive {ARCHIVE_PATH}...")
35
  with tarfile.open(ARCHIVE_PATH, "r:gz") as tar:
36
  tar.extractall(tmp_dir)
37
 
38
- # Get the extracted folder name (there should be only one)
39
  extracted_folder = os.path.join(tmp_dir, os.listdir(tmp_dir)[0])
40
 
41
- # Move to final destination
42
  os.rename(extracted_folder, TRANSFORMERS_REPOSITORY_PATH)
43
 
44
- # Remove the downloaded .tar.gz archive
45
  os.remove(ARCHIVE_PATH)
46
  print("Repository downloaded and extracted successfully.")
47
 
 
24
  HF_BASE_URL = "https://huggingface.co"
25
 
26
  if not os.path.exists(TRANSFORMERS_REPOSITORY_PATH):
 
 
27
  urllib.request.urlretrieve(TRANSFORMERS_REPOSITORY_URL, ARCHIVE_PATH)
28
 
 
29
  with tempfile.TemporaryDirectory() as tmp_dir:
 
 
30
  with tarfile.open(ARCHIVE_PATH, "r:gz") as tar:
31
  tar.extractall(tmp_dir)
32
 
 
33
  extracted_folder = os.path.join(tmp_dir, os.listdir(tmp_dir)[0])
34
 
 
35
  os.rename(extracted_folder, TRANSFORMERS_REPOSITORY_PATH)
36
 
 
37
  os.remove(ARCHIVE_PATH)
38
  print("Repository downloaded and extracted successfully.")
39