Update inference.py

inference.py

@@ -42,8 +42,8 @@ def generate_mask_for_feature(coor, raw_w, raw_h, mask=None):
 
     return coor_mask
 
-
-def infer_single_prompt(image_path, prompt, model_path, region=None, model_name="ferret_gemma", conv_mode="ferret_gemma_instruct"):
+## choose a conv_mode based on the model type from conversation.py
+def infer_single_prompt(image_path, prompt, model_path, region=None, model_name="ferret_llama", conv_mode="ferret_llama_3"):
     img = Image.open(image_path).convert('RGB')
 
     # this loads the model, image processor and tokenizer