update

app.py

@@ -48,9 +48,9 @@ def process_image(image, prompt):
     tuples, _ = processor.generate_tuples(prompt)
     dependencies, _ = processor.generate_dependencies(tuples)
     questions, _ = processor.generate_questions(
-        input_text, tuples.tuples, dependencies
+        prompt, tuples.tuples, dependencies
     )
-    reward = processor.get_reward(input_text, questions, dependencies, [image])
+    reward = processor.get_reward(questions, dependencies, [image])
     reward = reward[0]
     answers = {i: v > 0.5 for i, v in enumerate(reward)}
     graph_img = draw_colored_graph(dependencies, questions, answers)

scorer.py

@@ -322,7 +322,6 @@ Each tuple contains the following information:
     @spaces.GPU(duration=120)
     def get_reward(
         self,
-        prompt: str,
         questions: list[str],
         dependencies: dict[list],
         images: list[str],
@@ -331,7 +330,6 @@ Each tuple contains the following information:
         """Get reward for the generated questions use structured question graph.
 
         Args:
-            prompt (str): a prompt describe the image
             questions (list[str]): a list of questions generated based on the tuples
             dependencies (dict[list]): the dependencies between tuples
             images (list[str]): a list of image urls