Update app.py

app.py

@@ -17,12 +17,12 @@ def get_duration(
     image, model_selection
 ):
     width, height = image.size
-    megapixel = math.ceil((width * height) / 1_000_000)
+    pixel = width * height
 
     if model_selection in ["4xNomosWebPhotoRealPLKSR", "4xRealESRGAN"]:
-        return int(megapixel * 10)
+        return math.ceil((pixel * 10) / 1_000_000) + 3
     else:
-        return int(megapixel * 30)
+        return math.ceil((pixel * 30) / 1_000_000) + 3
 
 @spaces.GPU(duration = get_duration)
 def upscale_image(image, model_selection):