leekanghyeon commited on
Commit
64a93ab
1 Parent(s): ff449e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import requests
4
  from io import BytesIO
5
  import os
6
 
 
7
  api_token = os.getenv("HUGGINGFACE_API_TOKEN")
8
  if not api_token:
9
  st.error("API token not found. Please set the HUGGINGFACE_API_TOKEN environment variable.")
@@ -15,12 +16,14 @@ def transform_image(image):
15
  buffered = BytesIO()
16
  image.save(buffered, format="JPEG")
17
  buffered.seek(0)
18
- response = requests.post(api_url, headers=headers, files={"file": buffered})
 
 
19
 
20
  if response.status_code == 200:
21
  return Image.open(BytesIO(response.content))
22
  else:
23
- st.error("Failed to transform image.")
24
  return None
25
 
26
  st.title("AnimeGANv2 Image Transformer")
 
4
  from io import BytesIO
5
  import os
6
 
7
+ # 환경 변수에서 API 토큰을 가져옵니다.
8
  api_token = os.getenv("HUGGINGFACE_API_TOKEN")
9
  if not api_token:
10
  st.error("API token not found. Please set the HUGGINGFACE_API_TOKEN environment variable.")
 
16
  buffered = BytesIO()
17
  image.save(buffered, format="JPEG")
18
  buffered.seek(0)
19
+
20
+ files = {"file": buffered.getvalue()}
21
+ response = requests.post(api_url, headers=headers, files=files)
22
 
23
  if response.status_code == 200:
24
  return Image.open(BytesIO(response.content))
25
  else:
26
+ st.error(f"Failed to transform image. Status code: {response.status_code}. Response: {response.text}")
27
  return None
28
 
29
  st.title("AnimeGANv2 Image Transformer")