leekanghyeon commited on
Commit
ff449e8
1 Parent(s): 8713b43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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: