Spaces:
Sleeping
Sleeping
leekanghyeon
commited on
Commit
•
ff449e8
1
Parent(s):
8713b43
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,6 @@ import requests
|
|
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.")
|
@@ -17,7 +16,7 @@ def transform_image(image):
|
|
17 |
image.save(buffered, format="JPEG")
|
18 |
buffered.seek(0)
|
19 |
response = requests.post(api_url, headers=headers, files={"file": buffered})
|
20 |
-
|
21 |
if response.status_code == 200:
|
22 |
return Image.open(BytesIO(response.content))
|
23 |
else:
|
@@ -30,7 +29,7 @@ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png
|
|
30 |
if uploaded_file is not None:
|
31 |
image = Image.open(uploaded_file)
|
32 |
st.image(image, caption='Uploaded Image.', use_column_width=True)
|
33 |
-
|
34 |
if st.button("Transform"):
|
35 |
transformed_image = transform_image(image)
|
36 |
if transformed_image:
|
|
|
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.")
|
|
|
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:
|
|
|
29 |
if uploaded_file is not None:
|
30 |
image = Image.open(uploaded_file)
|
31 |
st.image(image, caption='Uploaded Image.', use_column_width=True)
|
32 |
+
|
33 |
if st.button("Transform"):
|
34 |
transformed_image = transform_image(image)
|
35 |
if transformed_image:
|