From e28fd3d3304429759e49771d608b66eceef21ae6 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 17 Feb 2025 16:24:14 +0100 Subject: [PATCH] added paths --- src/aiunn/inference.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/aiunn/inference.py b/src/aiunn/inference.py index 09b1098..b364191 100644 --- a/src/aiunn/inference.py +++ b/src/aiunn/inference.py @@ -131,6 +131,6 @@ class ImageUpscaler: return self.postprocess_image(output_tensor) # Example usage: -# upscaler = ImageUpscaler(model_path="/path/to/best_model") -# upscaled_image = upscaler.upscale_image("/path/to/low_res_image.jpg") -# upscaled_image.save("/path/to/upscaled_image.jpg") +upscaler = ImageUpscaler(model_path="/root/vision/aiuNN/best_model") +upscaled_image = upscaler.upscale_image("/root/vision/aiuNN/input.jpg") +upscaled_image.save("upscaled_image.jpg")