Update README.md

README.md

@@ -15,7 +15,7 @@ This version was exported using the old torch to ONNX converter (`torch.onnx.exp
 
 **Notes:**
 1. **Custom FourierUnitJIT**: A custom [FourierUnitJIT](https://github.com/Carve-Photos/lama/blob/main/saicinpainting/training/modules/ffc.py) implementation is used since the original cannot be directly ported to ONNX without overhead. The result is identical to the original model.
-2. **Fixed Input Shape**: The input shape is fixed at 1024x1024 pixels. Although dynamic input shapes are possible, they would require resolving issues with dynamic padding in the `irfft` and `rfftn` functions in `ffc.py`.
+2. **Fixed Input Shape**: The input shape is fixed at 800x800 pixels. Although dynamic input shapes are possible, they would require resolving issues with dynamic padding in the `irfft` and `rfftn` functions in `ffc.py`.
 3. **Opset Version 17**: This model uses opset version 17.
 
 ### 2. `lama.onnx` (NOT RECOMMENDED)
@@ -38,7 +38,11 @@ This version was exported using the new torch to ONNX converter (`torch.onnx.dyn
 **Original image:**
 ![original image](./image.jpg)
 
-**ONNX output:**
+
+**lama_fp32.onnx - output:**
+![onnx output](./output_onnx_fp32.png)
+
+**lama.onnx - output:**
 ![onnx output](./output_onnx.png)
 
 **Original model output:**