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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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 = os.getenv("HF_REPO_ID", "your_username/your_model")
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