Update README.md

README.md

@@ -70,7 +70,8 @@ final_output=jnp.argmax(outputs,axis=-1)
 Print(f"the predction of the dataset : {final_output}")
 ```
 
-Output:
-```The Correct answer: 1 ```
+```
+The Correct answer:-Option1 
+```