Update README.md

README.md

@@ -89,22 +89,18 @@ Use the code below to get started with the model.
 
 Install the Inference Code
 
-    !pip install hdm2 --quiet
+```bash
+pip install hdm2 --quiet
+```
 
 Run the HDM-2 model
 
+```python
     # Load the model from HuggingFace into the GPU
 
-<!---->
-
     from hdm2 import HallucinationDetectionModel
-
-<!---->
-
     hdm_model = HallucinationDetectionModel()
 
-<!---->
-
     prompt = "Explain how the heart functions"
     context = """
     The heart is a muscular organ that pumps blood throughout the body.
@@ -122,21 +118,19 @@ Run the HDM-2 model
     # average heart weighs ~10 oz (not 5 pounds),
     # and beats ~2.5 billion times (not 2 million) in a lifetime
 
-<!---->
-
     # Detect hallucinations with default parameters
     results = hdm_model.apply(prompt, context, response)
+```
 
 Print the results
 
+```python
     # Utility function to help with printing the model output
     def print_results(results):
      #print(results)
      # Print results
      print(f"\nHallucination severity: {results['adjusted_hallucination_severity']:.4f}")
 
-<!---->
-
      # Print hallucinated sentences
      if results['candidate_sentences']:
          print("\nPotentially hallucinated sentences:")
@@ -149,11 +143,10 @@ Print the results
            print("No hallucinated sentences detected.")
      else:
          print("\nNo hallucinated sentences detected.")
-
-<!---->
-
     print_results(results)
 
+```
+
 ```
 OUTPUT: