henok3878 commited on
Commit
b2d17dd
·
1 Parent(s): a55bf24

fix: add style files and fix their path

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. inference_utils.py +2 -2
Dockerfile CHANGED
@@ -17,6 +17,7 @@ COPY inference_utils.py .
17
  RUN mkdir -p packaged_models
18
 
19
  COPY ./packaged_models/* ./packaged_models/
 
20
 
21
  # Expose the port
22
  EXPOSE 7860
 
17
  RUN mkdir -p packaged_models
18
 
19
  COPY ./packaged_models/* ./packaged_models/
20
+ COPY ./styles/* ./styles/
21
 
22
  # Expose the port
23
  EXPOSE 7860
inference_utils.py CHANGED
@@ -83,7 +83,7 @@ def load_text(text_path: Union[Path, str]) -> str:
83
 
84
  def load_priming_data(style: int):
85
 
86
- priming_text = load_text(f"./data/samples/sample{style}.txt")
87
- priming_strokes = load_np_strokes(f"./data/samples/sample{style}.npy")
88
 
89
  return priming_text, priming_strokes
 
83
 
84
  def load_priming_data(style: int):
85
 
86
+ priming_text = load_text(f"./styles/style{style}.txt")
87
+ priming_strokes = load_np_strokes(f"./styles/style{style}.npy")
88
 
89
  return priming_text, priming_strokes