Hugging Face's logo

Spaces:
yeq6x
/
Image2Body_gradio
Running on Zero

App Files Community
yeq6x commited on
Commit
426f89e
·
1 Parent(s): 3e7d740

.

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. scripts/generate_prompt.py +1 -2
requirements.txt CHANGED
@@ -14,7 +14,7 @@ numpy==1.23.5
14
  opencv-python==4.9.0.80
15
  Pillow==10.3.0
16
  Requests==2.32.3
17
- tensorflow==2.16.1
18
  transforms==0.2.1
19
  tokenizers
20
  pytorch_lightning
 
14
  opencv-python==4.9.0.80
15
  Pillow==10.3.0
16
  Requests==2.32.3
17
+ tensorflow==2.18.0
18
  transforms==0.2.1
19
  tokenizers
20
  pytorch_lightning
scripts/generate_prompt.py CHANGED
@@ -8,7 +8,6 @@ import cv2
8
  import numpy as np
9
  from tensorflow.keras.layers import TFSMLayer
10
  from huggingface_hub import hf_hub_download
11
- from pathlib import Path
12
 
13
  # 画像サイズの設定
14
  IMAGE_SIZE = 448
@@ -46,7 +45,7 @@ def load_wd14_tagger_model():
46
  download_model_files(DEFAULT_REPO, model_dir, VAR_DIR, MODEL_FILES, VAR_FILES)
47
  else:
48
  print("Using existing model")
49
- # return TFSMLayer(model_dir, call_endpoint='serving_default')
50
 
51
  def read_tags_from_csv(csv_path):
52
  """CSVファイルからタグを読み取る"""
 
8
  import numpy as np
9
  from tensorflow.keras.layers import TFSMLayer
10
  from huggingface_hub import hf_hub_download
 
11
 
12
  # 画像サイズの設定
13
  IMAGE_SIZE = 448
 
45
  download_model_files(DEFAULT_REPO, model_dir, VAR_DIR, MODEL_FILES, VAR_FILES)
46
  else:
47
  print("Using existing model")
48
+ return TFSMLayer(model_dir, call_endpoint='serving_default')
49
 
50
  def read_tags_from_csv(csv_path):
51
  """CSVファイルからタグを読み取る"""