Update README.md

README.md

@@ -10,10 +10,6 @@ WebGen-Agent is an advanced website generation agent designed to autonomously cr
 
 WebGen-Agent combines state-of-the-art language models with specialized training techniques to create a powerful website generation tool. The agent can understand natural language instructions specifying appearance and functional requirements, iteratively generate website codebases, and refine them using visual and functional feedback.
 
-![WebGen-Agent Workflow](fig/webgen-agent.png)
-
-![Step-GRPO with Screenshot and GUI-agent Feedback](fig/step-grpo.png)
-
 ## Resources
 
 Links to the data and model parameters are as follows:
@@ -42,12 +38,7 @@ WebGen-Agent follows an iterative, multi-step paradigm for website generation:
    - A GUI-agent tests the website functionality and provides functional feedback
 4. **Refinement**: Based on the feedback, the agent continues to improve the website until it meets requirements
 
-## Key Features
-
-- **Iterative Refinement**: Continuously improves website appearance and functionality
-- **Feedback Integration**: Uses both visual and functional feedback for enhanced performance
-- **Backtracking Mechanism**: Reverts to previous states when encountering persistent errors
-- **Best Step Selection**: Selects the optimal version based on screenshot and GUI-agent scores
+![WebGen-Agent Workflow](fig/webgen-agent.png)
 
 ## Step-GRPO with Screenshot and GUI-agent Feedback
 
@@ -57,6 +48,8 @@ The Step-GRPO with Screenshot and GUI-agent Feedback approach uses the screensho
 
 These dual rewards provide dense, reliable process supervision that significantly improves the model's ability to generate high-quality websites.
 
+![Step-GRPO with Screenshot and GUI-agent Feedback](fig/step-grpo.png)
+
 ## Citation
 
 If you find our project useful, please cite: