small python fix

detect-pretrain-code-contamination/src/run.py

@@ -44,7 +44,7 @@ def load_model(name1):
     if name1 not in models:
         model1 = AutoModelForCausalLM.from_pretrained(name1, return_dict=True, device_map='auto')
         model1.eval()
-        if name1.contains('mistral') or name1.contains('Mistral'): #Loading default mistral tokenizers as some tokenizers don't work out of the box.
+        if "mistral" in name1 or "Mistral" in name1: #Loading default mistral tokenizers as some tokenizers don't work out of the box.
             tokenizer1 = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
         else:
             tokenizer1 = AutoTokenizer.from_pretrained(name1)