Update main.py

main.py

@@ -28,18 +28,6 @@ def generate_image_from_text(prompt):
     else:
         r.raise_for_status()
 
-def resize_image(image_bytes, max_size=(256, 256)):
-    # Open the image from bytes
-    img = Image.open(BytesIO(image_bytes))
-    
-    # Resize the image
-    img.thumbnail(max_size)
-    
-    # Save it back to bytes
-    buffer = BytesIO()
-    img.save(buffer, format="PNG")
-    return buffer.getvalue()
-
 def upscale_image_esrgan(image_bytes):
     # Set up environment variables
     os.environ['ESRGAN_API_KEY'] = ESRGAN_API_KEY
@@ -93,11 +81,8 @@ def main():
         st.success("Generating image from text prompt...")
         image_bytes = generate_image_from_text(prompt)
         
-        st.success("Resizing image...")
-        resized_image_bytes = resize_image(image_bytes)
-        
         st.success("Upscaling image with ESRGAN...")
-        upscaled_image_bytes = upscale_image_esrgan(resized_image_bytes)
+        upscaled_image_bytes = upscale_image_esrgan(image_bytes)
         
         st.success("Further upscaling image with GFPGAN...")
         img = further_upscale_image(upscaled_image_bytes)