Commit 0c8771d8 authored by jameskrw's avatar jameskrw
Browse files

udpated prompt

parent 6fce76ea
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -137,7 +137,8 @@ class CrossViewEnv(BaseEnv):
    def system_prompt(self) -> str:
        return """You are an AI assistant that answers visual questions based on images.
Given images and a question, first give your thought then answer.
Your answer should be in the format of <think>...</think><answer>...</answer>."""
Your answer should be in the format of <think>...</think><answer>...</answer>.
e.g. <think>I can see there're multiple images with different view. I can see from the second view the object is on the target's left.I think the correct answer is A</think><answer>A</answer>"""
    
    def compute_reward(self) -> float:
        """Return the total reward accumulated so far"""