Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from typing import List
|
|
| 7 |
|
| 8 |
# Download model from Hugging Face Hub
|
| 9 |
def download_model():
|
| 10 |
-
REPO_ID =
|
| 11 |
FILENAMES = [
|
| 12 |
"checkpoints/model-17900.data-00000-of-00001",
|
| 13 |
"checkpoints/model-17900.index",
|
|
@@ -20,8 +20,7 @@ def download_model():
|
|
| 20 |
hf_hub_download(
|
| 21 |
repo_id=REPO_ID,
|
| 22 |
filename=filename,
|
| 23 |
-
local_dir=".",
|
| 24 |
-
token=os.getenv("HF_TOKEN")
|
| 25 |
)
|
| 26 |
|
| 27 |
# Download model on startup
|
|
|
|
| 7 |
|
| 8 |
# Download model from Hugging Face Hub
|
| 9 |
def download_model():
|
| 10 |
+
REPO_ID = "3rrormr/handwriting-synthesis-api" # Use your actual repo name
|
| 11 |
FILENAMES = [
|
| 12 |
"checkpoints/model-17900.data-00000-of-00001",
|
| 13 |
"checkpoints/model-17900.index",
|
|
|
|
| 20 |
hf_hub_download(
|
| 21 |
repo_id=REPO_ID,
|
| 22 |
filename=filename,
|
| 23 |
+
local_dir=".", # or local_dir="checkpoints", up to you) # Or leave out if public
|
|
|
|
| 24 |
)
|
| 25 |
|
| 26 |
# Download model on startup
|