cocoat commited on
Commit
c606fb9
·
verified ·
1 Parent(s): 62c88a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -46,37 +46,37 @@ api = HfApi()
46
  # 画像をアップロードするリポジトリID
47
  HF_REPO_ID = "cocoat/Re.cocoamixXL3"
48
 
 
49
  def upload_image_to_hub(image_pil, prompt_text):
50
- # ファイル名を生成(タイムスタンプとプロンプトの一部)
51
  timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
52
- # プロンプトから安全なファイル名の一部を生成
53
  safe_prompt = "".join(c for c in prompt_text if c.isalnum() or c in (' ', '.', '_')).replace(' ', '_')[:30]
54
  filename = f"image_{timestamp}_{safe_prompt}.png"
55
  filepath = f"temp_{filename}" # 一時的にローカルに保存するパス
56
- # PIL画像を一時ファイルとして保存
57
  image_pil.save(filepath)
58
 
59
- # Hubにアップロード
60
  try:
61
- # リポジトリ内にディレクトリを作成する場合は path_in_repo を使う
62
  path_in_repo = f"generated_images/{filename}"
63
  upload_info = api.upload_file(
64
- path_or_fileobj=filepath,
65
- path_in_repo=path_in_repo,
66
- repo_id=HF_REPO_ID,
67
- repo_type="dataset", # または "space", "model" など、目的のリポジトリタイプ
68
- # 通常、画像保存には "dataset" タイプのリポジトリが適しています
69
  )
70
  print(f"Uploaded {filepath} to {upload_info.url}")
71
  return upload_info.url # アップロードされたファイルのURLを返す
72
- except Exception as e:
73
  print(f"Error uploading image to Hub: {e}")
74
  return None
75
  finally:
76
- # 一時ファイルを削除
77
  if os.path.exists(filepath):
78
- os.remove(filepath)
79
-
80
 
81
  def make_html_table(caption):
82
  rows = caption.split("\n")
 
46
  # 画像をアップロードするリポジトリID
47
  HF_REPO_ID = "cocoat/Re.cocoamixXL3"
48
 
49
+
50
  def upload_image_to_hub(image_pil, prompt_text):
51
+ # ファイル名を生成(タイムスタンプとプロンプトの一部)
52
  timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
53
+ # プロンプトから安全なファイル名の一部を生成
54
  safe_prompt = "".join(c for c in prompt_text if c.isalnum() or c in (' ', '.', '_')).replace(' ', '_')[:30]
55
  filename = f"image_{timestamp}_{safe_prompt}.png"
56
  filepath = f"temp_{filename}" # 一時的にローカルに保存するパス
57
+ # PIL画像を一時ファイルとして保存
58
  image_pil.save(filepath)
59
 
60
+ # Hubにアップロード
61
  try:
62
+ # リポジトリ内にディレクトリを作成する場合は path_in_repo を使う
63
  path_in_repo = f"generated_images/{filename}"
64
  upload_info = api.upload_file(
65
+ path_or_fileobj=filepath,
66
+ path_in_repo=path_in_repo,
67
+ repo_id=HF_REPO_ID,
68
+ repo_type="dataset", # または "space", "model" など、目的のリポジトリタイプ
69
+ # 通常、画像保存には "dataset" タイプのリポジトリが適しています
70
  )
71
  print(f"Uploaded {filepath} to {upload_info.url}")
72
  return upload_info.url # アップロードされたファイルのURLを返す
73
+ except Exception as e:
74
  print(f"Error uploading image to Hub: {e}")
75
  return None
76
  finally:
77
+ # 一時ファイルを削除
78
  if os.path.exists(filepath):
79
+ os.remove(filepath)
 
80
 
81
  def make_html_table(caption):
82
  rows = caption.split("\n")