diff --git a/scrapegraphai/nodes/generate_answer_from_image_node.py b/scrapegraphai/nodes/generate_answer_from_image_node.py
index 4cc93d18..7134cabe 100644
--- a/scrapegraphai/nodes/generate_answer_from_image_node.py
+++ b/scrapegraphai/nodes/generate_answer_from_image_node.py
@@ -73,7 +73,7 @@ class GenerateAnswerFromImageNode(BaseNode):
 
         supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo")
 
-        if self.node_config["config"]["llm"]["model"] not in supported_models:
+        if self.node_config["config"]["llm"]["model"].split("/")[-1]not in supported_models:
             raise ValueError(f"""Model '{self.node_config['config']['llm']['model']}'
                              is not supported. Supported models are: 
                              {', '.join(supported_models)}.""")
diff --git a/scrapegraphai/nodes/merge_answers_node.py b/scrapegraphai/nodes/merge_answers_node.py
index 1e25cccb..ff27d7b3 100644
--- a/scrapegraphai/nodes/merge_answers_node.py
+++ b/scrapegraphai/nodes/merge_answers_node.py
@@ -76,7 +76,6 @@ class MergeAnswersNode(BaseNode):
                 self.llm_model = self.llm_model.with_structured_output(
                     schema = self.node_config["schema"],
                     method="function_calling") # json schema works only on specific models
-                
                 # default parser to empty lambda function
                 output_parser = lambda x: x
                 if is_basemodel_subclass(self.node_config["schema"]):