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")
+upscaler = ImageUpscaler(model_path="/root/vision/aiuNN/best_model")
-# upscaled_image = upscaler.upscale_image("/path/to/low_res_image.jpg")
+upscaled_image = upscaler.upscale_image("/root/vision/aiuNN/input.jpg")
-# upscaled_image.save("/path/to/upscaled_image.jpg")
+upscaled_image.save("upscaled_image.jpg")