Update main.py

main.py

@@ -27,7 +27,7 @@ def generate_image_from_text(prompt):
     else:
         r.raise_for_status()
 
-def resize_image(image_bytes, max_size=(1024, 1024)):
+def resize_image(image_bytes, max_size=(640, 640)):
     # Open the image from bytes
     img = Image.open(BytesIO(image_bytes))
     
@@ -105,4 +105,4 @@ def main():
         st.image(img, caption='Upscaled Image', use_column_width=True)
 
 if __name__ == "__main__":
-    main()
+    main()