Update app.py

app.py

@@ -1258,11 +1258,9 @@ def handle_generation(h,w,d):
         difficulty_points + 1
 
     pxs = h*w
-    if pxs > 4 * (10 ** 6):
-        difficulty_points + 3
-    elif pxs > 3 * (10 ** 6):
+    if pxs > 2 * (10 ** 6):
         difficulty_points + 2
-    elif pxs > 2 * (10 ** 6):
+    elif pxs > 1 * (10 ** 6):
         difficulty_points + 1
 
     if difficulty_points < 2:
@@ -1276,11 +1274,11 @@ def handle_generation(h,w,d):
 def easy_generation(h,w,d):
     return generation(h,w,d)
 
-@spaces.GPU(duration=180)
+@spaces.GPU(duration=210)
 def balanced_generation(h,w,d):
     return generation(h,w,d)
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=270)
 def hard_generation(h,w,d):
     return generation(h,w,d)
 
@@ -1343,7 +1341,7 @@ if __name__ == "__main__":
                 height = gr.Slider(
                     label="Height (px)",
                     minimum=512,
-                    maximum=2048,
+                    maximum=1536,
                     step=16,
                     value=1024,
                 )
@@ -1351,7 +1349,7 @@ if __name__ == "__main__":
                 width = gr.Slider(
                     label="Width (px)",
                     minimum=512,
-                    maximum=2048,
+                    maximum=1536,
                     step=16,
                     value=1024,
                 )
@@ -1390,7 +1388,7 @@ if __name__ == "__main__":
             outputs=[cover]
         )
         upscale_now.click(
-            fn=upscaler,
+            fn=handle_upscaler,
             inputs=[cover],
             outputs=[cover]
         )