Update README.md

README.md

@@ -80,7 +80,7 @@ Prompt:
 Write a simple multithreaded web server in Rust that serves "Hello, world!" to any GET request.
 
 HydraCoder Output:
-
+```rust
 use std::net::{TcpListener, TcpStream};
 use std::io::{Read, Write};
 use std::thread;
@@ -110,14 +110,14 @@ fn main() -> std::io::Result<()> {
 
     Ok(())
 }
-
+```
 
 ---
 
 ⚡ Inference Code
 
 You can run inference using transformers and text-generation pipeline:
-
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 model_id = "Daemontatox/HydraCoder"
@@ -131,7 +131,7 @@ prompt = "Write a function in Rust that takes a list of integers and returns the
 
 output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.2)[0]["generated_text"]
 print(output)
-
+```
 
 ---