.

scripts/generate_prompt.py

@@ -1,16 +1,16 @@
-import argparse
 import csv
 import os
-import json
-
-from PIL import Image
 import cv2
 import numpy as np
+import tensorflow as tf
 from tensorflow.keras.layers import TFSMLayer
 from huggingface_hub import hf_hub_download
 
 import spaces
 
+# TensorFlowがGPUを使用しないように設定
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
 # 画像サイズの設定
 IMAGE_SIZE = 448