Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,8 +7,9 @@ from typing import List
|
|
| 7 |
|
| 8 |
# Download model from Hugging Face Hub
|
| 9 |
def download_model():
|
| 10 |
-
REPO_ID = "3rrormr/handwriting-synthesis-api" #
|
| 11 |
FILENAMES = [
|
|
|
|
| 12 |
"checkpoints/model-17900.data-00000-of-00001",
|
| 13 |
"checkpoints/model-17900.index",
|
| 14 |
"checkpoints/model-17900.meta"
|
|
@@ -20,9 +21,10 @@ def download_model():
|
|
| 20 |
hf_hub_download(
|
| 21 |
repo_id=REPO_ID,
|
| 22 |
filename=filename,
|
| 23 |
-
local_dir=".",
|
| 24 |
)
|
| 25 |
|
|
|
|
| 26 |
# Download model on startup
|
| 27 |
try:
|
| 28 |
download_model()
|
|
|
|
| 7 |
|
| 8 |
# Download model from Hugging Face Hub
|
| 9 |
def download_model():
|
| 10 |
+
REPO_ID = "3rrormr/handwriting-synthesis-api" # Your model repo
|
| 11 |
FILENAMES = [
|
| 12 |
+
"checkpoints/checkpoint",
|
| 13 |
"checkpoints/model-17900.data-00000-of-00001",
|
| 14 |
"checkpoints/model-17900.index",
|
| 15 |
"checkpoints/model-17900.meta"
|
|
|
|
| 21 |
hf_hub_download(
|
| 22 |
repo_id=REPO_ID,
|
| 23 |
filename=filename,
|
| 24 |
+
local_dir=".", # or local_dir="checkpoints", up to you
|
| 25 |
)
|
| 26 |
|
| 27 |
+
|
| 28 |
# Download model on startup
|
| 29 |
try:
|
| 30 |
download_model()
|