tomo2chin2 commited on
Commit
0683a99
·
verified ·
1 Parent(s): 2d6e5b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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}.png"
68
  path_in_repo = f"images/{filename}"
69
 
70
- # 一時ファイルに保存
71
- with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
72
  tmp_path = tmp_file.name
73
- image.save(tmp_path, format="PNG")
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