Spaces:
Runtime error
Runtime error
- requirements.txt +1 -1
- 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.
|
| 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 |
-
|
| 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ファイルからタグを読み取る"""
|