Spaces:
Runtime error
Runtime error
fahmiaziz98
commited on
Commit
·
32814be
1
Parent(s):
ccd3131
frist commit
Browse files- requirements.txt +1 -0
- scripts/s3.py +0 -1
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
boto3
|
| 2 |
torch
|
|
|
|
| 3 |
transformers
|
| 4 |
fastapi
|
| 5 |
uvicorn[standard]
|
|
|
|
| 1 |
boto3
|
| 2 |
torch
|
| 3 |
+
pillow
|
| 4 |
transformers
|
| 5 |
fastapi
|
| 6 |
uvicorn[standard]
|
scripts/s3.py
CHANGED
|
@@ -30,7 +30,6 @@ def download_model_from_s3(local_path: str, s3_prefix: str):
|
|
| 30 |
local_file = os.path.join(local_path, os.path.relpath(s3_key, s3_prefix))
|
| 31 |
|
| 32 |
os.makedirs(os.path.dirname(local_file), exist_ok=True)
|
| 33 |
-
logger.info(f"Downloading model from {BUCKET_NAME} key {s3_key} to {local_file}")
|
| 34 |
s3.download_file(BUCKET_NAME, s3_key, local_file)
|
| 35 |
logger.info(f"Completed download {s3_key} to {local_file}")
|
| 36 |
except Exception as e:
|
|
|
|
| 30 |
local_file = os.path.join(local_path, os.path.relpath(s3_key, s3_prefix))
|
| 31 |
|
| 32 |
os.makedirs(os.path.dirname(local_file), exist_ok=True)
|
|
|
|
| 33 |
s3.download_file(BUCKET_NAME, s3_key, local_file)
|
| 34 |
logger.info(f"Completed download {s3_key} to {local_file}")
|
| 35 |
except Exception as e:
|