Hugging Face's logo

Spaces:
yeq6x
/
Image2Body_gradio
Running on Zero

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

.

Browse files
Files changed (1) hide show
  1. scripts/generate_prompt.py +1 -1
scripts/generate_prompt.py CHANGED
@@ -46,7 +46,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ファイルからタグを読み取る"""
 
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ファイルからタグを読み取る"""