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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -59,22 +59,22 @@ def upload_image_to_hub(image_pil, prompt_text):
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
 
 
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