Update app.py

app.py

@@ -19,6 +19,7 @@ if not torch.cuda.is_available():
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv(
     'CACHE_EXAMPLES') == '1'
+MAX_IMAGE_SIZE = int(os.getenv('MAX_IMAGE_SIZE', '2048'))
 USE_TORCH_COMPILE = os.getenv('USE_TORCH_COMPILE') == '1'
 ENABLE_CPU_OFFLOAD = os.getenv('ENABLE_CPU_OFFLOAD') == '1'
 
@@ -158,14 +159,14 @@ with gr.Blocks(css='style.css') as demo:
                 width = gr.Slider(
                     label='Width',
                     minimum=256,
-                    maximum=4096,
+                    maximum=MAX_IMAGE_SIZE,
                     step=32,
                     value=1024,
                 )
                 height = gr.Slider(
                     label='Height',
                     minimum=256,
-                    maximum=4096,
+                    maximum=MAX_IMAGE_SIZE,
                     step=32,
                     value=1024,
                 )