simonlee-cb commited on
Commit
97b9bcc
·
1 Parent(s): 1c74bd7

feat: resize image before uploading

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -15,6 +15,12 @@ def gallery(column, images):
15
  for i, image in enumerate(group):
16
  cols[i].image(image)
17
 
 
 
 
 
 
 
18
  st.title('CollageAI')
19
 
20
  user_images = None
@@ -41,7 +47,8 @@ if submitted:
41
  with st.container():
42
  with st.spinner('Uploading images...'):
43
  try:
44
- uploaded_images = client.upload_images(user_images)
 
45
  except Exception as e:
46
  st.error(f"Error uploading images: {e}")
47
 
 
15
  for i, image in enumerate(group):
16
  cols[i].image(image)
17
 
18
+ def resize_image(image_file):
19
+ image = Image.open(image_file)
20
+ resized = image.resize((224, 224))
21
+ resized.save(image_file.name)
22
+ return open(image_file.name, 'rb')
23
+
24
  st.title('CollageAI')
25
 
26
  user_images = None
 
47
  with st.container():
48
  with st.spinner('Uploading images...'):
49
  try:
50
+ resized_images = [resize_image(image) for image in user_images]
51
+ uploaded_images = client.upload_images(resized_images)
52
  except Exception as e:
53
  st.error(f"Error uploading images: {e}")
54