Update app.py

app.py

@@ -20,8 +20,9 @@ client = genai.Client(api_key=api_key)
 model = torch.hub.load('pytorch/vision:v0.10.0', 'srcnn', pretrained=True)
 model.eval()
 
-def upscale_image(image):
-    """Nâng cấp độ phân giải ảnh bằng SRCNN"""
+def upscale_image(image, target_resolution=(2560, 1440)):
+    """Nâng cấp độ phân giải ảnh bằng nội suy trước khi qua SRCNN"""
+    image = image.resize(target_resolution, Image.BICUBIC)  # Nội suy trước khi SRCNN xử lý
     transform = transforms.Compose([
         transforms.ToTensor(),
         transforms.Lambda(lambda x: x.unsqueeze(0))  # Thêm batch dimension