Spaces:
Sleeping
Sleeping
leekanghyeon
commited on
Commit
•
64a93ab
1
Parent(s):
ff449e8
Update app.py
Browse files
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 |
-
|
|
|
|
|
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")
|