Update app.py
Browse files
app.py
CHANGED
|
@@ -61,16 +61,16 @@ class HuggingFaceUploader:
|
|
| 61 |
logger.error("HF_TOKENが設定されていないため、アップロードできません")
|
| 62 |
return None
|
| 63 |
|
| 64 |
-
# ユニークなファイル名を生成
|
| 65 |
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 66 |
unique_id = str(uuid.uuid4())[:8]
|
| 67 |
-
filename = f"{prefix}_{timestamp}_{unique_id}.
|
| 68 |
path_in_repo = f"images/{filename}"
|
| 69 |
|
| 70 |
-
# 一時ファイルに保存
|
| 71 |
-
with tempfile.NamedTemporaryFile(suffix=".
|
| 72 |
tmp_path = tmp_file.name
|
| 73 |
-
image.save(tmp_path, format="
|
| 74 |
|
| 75 |
logger.info(f"画像を一時ファイルに保存: {tmp_path}")
|
| 76 |
|
|
@@ -89,7 +89,7 @@ class HuggingFaceUploader:
|
|
| 89 |
except Exception as e:
|
| 90 |
logger.warning(f"一時ファイル削除エラー: {e}")
|
| 91 |
|
| 92 |
-
# URLを構築して返す
|
| 93 |
url = f"https://huggingface.co/datasets/{self.repo_id}/resolve/main/{path_in_repo}"
|
| 94 |
logger.info(f"アップロード成功: {url}")
|
| 95 |
return url
|
|
|
|
| 61 |
logger.error("HF_TOKENが設定されていないため、アップロードできません")
|
| 62 |
return None
|
| 63 |
|
| 64 |
+
# ユニークなファイル名を生成 - JPEGフォーマットを使用
|
| 65 |
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 66 |
unique_id = str(uuid.uuid4())[:8]
|
| 67 |
+
filename = f"{prefix}_{timestamp}_{unique_id}.jpg"
|
| 68 |
path_in_repo = f"images/{filename}"
|
| 69 |
|
| 70 |
+
# 一時ファイルに保存 - JPEGフォーマットで保存
|
| 71 |
+
with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp_file:
|
| 72 |
tmp_path = tmp_file.name
|
| 73 |
+
image.save(tmp_path, format="JPEG", quality=95) # 高品質JPEGとして保存
|
| 74 |
|
| 75 |
logger.info(f"画像を一時ファイルに保存: {tmp_path}")
|
| 76 |
|
|
|
|
| 89 |
except Exception as e:
|
| 90 |
logger.warning(f"一時ファイル削除エラー: {e}")
|
| 91 |
|
| 92 |
+
# URLを構築して返す - データセットリポジトリの場合は datasets/ を含める
|
| 93 |
url = f"https://huggingface.co/datasets/{self.repo_id}/resolve/main/{path_in_repo}"
|
| 94 |
logger.info(f"アップロード成功: {url}")
|
| 95 |
return url
|