Update simple example with Ollama

Author: hermanzdosilovic

README.md

@@ -246,8 +246,12 @@ client = Client(
     headers={"Authorization": "Bearer " + os.environ.get("OLLAMA_API_KEY")},
 )
 
-system = "You are a helpful assistant that can execute Python code. Only respond with the code to be executed and nothing else. Strip backticks in code blocks."
-prompt = "Calculate how many r's are in the word 'strawberry'."
+system = """
+You are a helpful assistant that can execute code written in the C programming language.
+Only respond with the code written in the C programming language that needs to be executed and nothing else.
+Strip the backticks in code blocks.
+"""
+prompt = "How many r's are in the word 'strawberry'."
 
 response = client.chat(
     model="gpt-oss:120b-cloud",
@@ -258,10 +262,10 @@ response = client.chat(
 )
 
 code = response["message"]["content"]
-print(f"Generated code:\n{code}")
+print(f"CODE GENERATED BY THE MODEL:\n{code}\n")
 
-result = judge0.run(source_code=code, language=judge0.PYTHON)
-print(f"Execution result:\n{result.stdout}")
+result = judge0.run(source_code=code, language=judge0.C)
+print(f"CODE EXECUTION RESULT:\n{result.stdout}")
 ```
 
 #### Tool Calling (a.k.a. Function Calling) with Ollama