Update README.md

README.md

@@ -88,12 +88,13 @@ from embedl.models.vllm import LLM
 
 model_id = "embedl/Llama-3.2-1B-Instruct-FlashHead"
 
-sampling = SamplingParams(max_tokens=128, temperature=0.0)
-llm = LLM(model=model_id, trust_remote_code=True)
-
-prompt = "Write a haiku about coffee."
-output = llm.generate([prompt], sampling)
-print(output[0].outputs[0].text)
+if __name__ == "__main__":
+    sampling = SamplingParams(max_tokens=128, temperature=0.0)
+    llm = LLM(model=model_id, trust_remote_code=True)
+    
+    prompt = "Write a haiku about coffee."
+    output = llm.generate([prompt], sampling)
+    print(output[0].outputs[0].text)
 ```
 
 ---
@@ -108,11 +109,13 @@ import asyncio
 from embedl.models.vllm.demo import run_repl
 
 model_id = "embedl/Llama-3.2-1B-Instruct-FlashHead"
-asyncio.run(
-    run_repl(
-        model=model_id
+
+if __name__ == "__main__":
+    asyncio.run(
+        run_repl(
+            model=model_id
+        )
     )
-)
 ```
 ---