Spaces:

luca115
/

best_upscaling_models

Running on Zero

luca115 commited on 17 days ago

Commit

7b74cf4

verified ·

1 Parent(s): 23553ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ MODELS = {
 def get_duration(
     image, model_selection
 ):
-    height, width, _ = image.shape
     megapixel = math.ceil((width * height) / 1_000_000)
     if model_selection in ["4xNomosWebPhotoRealPLKSR", "4xRealESRGAN"]:

 def get_duration(
     image, model_selection
 ):
+    width, height = image.size
     megapixel = math.ceil((width * height) / 1_000_000)
     if model_selection in ["4xNomosWebPhotoRealPLKSR", "4xRealESRGAN"]: