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")