3morrrrr commited on
Commit
bf4f6ee
·
verified ·
1 Parent(s): d948cab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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" # Use your actual repo name
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=".", # or local_dir="checkpoints", up to you) # Or leave out if public
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()