Update README.md

README.md

@@ -26,7 +26,9 @@ Yi-VL adopts the [LLaVA](https://github.com/haotian-liu/LLaVA) architecture, whi
 
 This has been implemented into the SGLang codebase, where you can simply call this model by creating a function like so:
 ```
-sgl.function
+import sglang as sgl
+
+@sgl.function
 def image_qa(s, image_path, question):
     s += sgl.user(sgl.image(image_path) + question)
     s += sgl.assistant(sgl.gen("answer"))