Update app.py
Browse files
app.py
CHANGED
|
@@ -24,29 +24,12 @@ def process_image(input_image):
|
|
| 24 |
if input_image is None:
|
| 25 |
return None
|
| 26 |
|
| 27 |
-
# Ensure input_image is a numpy array
|
| 28 |
-
input_array = np.array(input_image)
|
| 29 |
-
|
| 30 |
# Convert to PIL Image for resizing
|
| 31 |
-
pil_image = Image.fromarray(
|
| 32 |
-
|
| 33 |
-
# Resize the longest side to 256 while maintaining aspect ratio
|
| 34 |
-
width, height = pil_image.size
|
| 35 |
-
if width > height:
|
| 36 |
-
new_width = 256
|
| 37 |
-
new_height = int(height * (256 / width))
|
| 38 |
-
else:
|
| 39 |
-
new_height = 256
|
| 40 |
-
new_width = int(width * (256 / height))
|
| 41 |
-
|
| 42 |
-
resized_image = pil_image.resize((new_width, new_height), Image.LANCZOS)
|
| 43 |
-
|
| 44 |
-
# Convert back to numpy array
|
| 45 |
-
resized_array = np.array(resized_image)
|
| 46 |
|
| 47 |
# Upscale the image using AuraSR
|
| 48 |
with torch.no_grad():
|
| 49 |
-
upscaled_image = aura_sr.upscale_4x(
|
| 50 |
|
| 51 |
# Convert result to numpy array if it's not already
|
| 52 |
result_array = np.array(upscaled_image)
|
|
|
|
| 24 |
if input_image is None:
|
| 25 |
return None
|
| 26 |
|
|
|
|
|
|
|
|
|
|
| 27 |
# Convert to PIL Image for resizing
|
| 28 |
+
pil_image = Image.fromarray(input_image)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# Upscale the image using AuraSR
|
| 31 |
with torch.no_grad():
|
| 32 |
+
upscaled_image = aura_sr.upscale_4x(pil_image)
|
| 33 |
|
| 34 |
# Convert result to numpy array if it's not already
|
| 35 |
result_array = np.array(upscaled_image)
|