Update README.md

README.md

@@ -65,18 +65,9 @@ image_sizes = [[image.size[0], image.size[1]]]
 prompt = "In the picture shown below, prove ΔWXY and ΔZWY are similar. Please conclude your answer as Answer: xxx at the end if possible."
 
 # Set up conversation template
-try:
-    conv_template = "qwen_2_5"
-    conv = copy.deepcopy(conv_templates[conv_template])
-except KeyError:
-    available_templates = list(conv_templates.keys())
-    for template_name in available_templates:
-        if 'qwen' in template_name.lower():
-            conv_template = template_name
-            break
-    else:
-        conv_template = available_templates[0]
-    conv = copy.deepcopy(conv_templates[conv_template])
+
+conv_template = "qwen_2_5"
+conv = copy.deepcopy(conv_templates[conv_template])
 
 # Add question with image
 question = DEFAULT_IMAGE_TOKEN + "\n" + prompt