From fee77d1fe6e9f55d2fe0082274a6d01b952e552f Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Mon, 19 Aug 2024 14:35:13 +0200
Subject: [PATCH] Update screenshot_scraper_graph.py

---
 scrapegraphai/graphs/screenshot_scraper_graph.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scrapegraphai/graphs/screenshot_scraper_graph.py b/scrapegraphai/graphs/screenshot_scraper_graph.py
index 13046d93..f3ce608d 100644
--- a/scrapegraphai/graphs/screenshot_scraper_graph.py
+++ b/scrapegraphai/graphs/screenshot_scraper_graph.py
@@ -48,8 +48,8 @@ class ScreenshotScraperGraph(AbstractGraph):
             }
         )
         generate_answer_from_image_node = GenerateAnswerFromImageNode(
-            input="imgs",
-            output=["screenshots"],
+            input="screenshots",
+            output=["answer"],
             node_config={
                 "config": self.config
             }