updatge

app.py

@@ -54,7 +54,7 @@ def process_image(image, prompt):
     reward = reward[0]
     answers = {str(i): v > 0.5 for i, v in enumerate(reward)}
     graph_img = draw_colored_graph(dependencies, questions, answers)
-    return reward, f"""
+    return graph_img, f"""
 Question: {questions}.
 Reward per question: {reward}"""
 

scorer.py

@@ -319,6 +319,7 @@ Each tuple contains the following information:
 
         return sorted_questions
 
+    @spaces.GPU()
     def get_reward(
         self,
         questions: list[str],
@@ -333,6 +334,9 @@ Each tuple contains the following information:
             dependencies (dict[list]): the dependencies between tuples
             images (list[str]): a list of image urls
         """
+
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.binary_vqa.to(self.device)
         scores = {}
 
         sorted_questions = self._create_graph_questions(questions, dependencies)