roli253 commited on
Commit
c90edca
·
verified ·
1 Parent(s): 0e88f46

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -78
README.md CHANGED
@@ -1,78 +1,16 @@
1
- ---
2
- license: apache-2.0
3
- base_model: Qwen/Qwen3-4B-Instruct-2507
4
- tags:
5
- - fitness
6
- - pose-correction
7
- - exercise-coaching
8
- - fine-tuned
9
- language:
10
- - en
11
- - hu
12
- ---
13
-
14
- # PoseFit Correction Model
15
-
16
- Fine-tuned Qwen3-4B model for generating real-time exercise correction commands based on pose analysis.
17
-
18
- ## Model Description
19
-
20
- This model analyzes exercise movement data (angles, phases, orientation) and provides short, actionable correction commands in English and Hungarian.
21
-
22
- **Base Model:** Qwen/Qwen3-4B-Instruct-2507
23
- **Fine-tuning:** LoRA (merged)
24
- **Training:** ~700 synthetic samples across 11 exercises
25
-
26
- ## Supported Exercises
27
-
28
- - Calf Raises
29
- - Glute Bridges
30
- - Heel Touches
31
- - Jumping Jacks
32
- - Lunges
33
- - Mountain Climbers
34
- - Plank
35
- - Pushups
36
- - Russian Twist
37
- - Sit-ups
38
- - Squat
39
-
40
- ## Input Format
41
-
42
- ```
43
- Mean Actual Angle: 133.8 degrees
44
- Mean Target Angle: 171.6 degrees
45
- Joint: left_shoulder-left_hip-left_knee
46
- Phase: throughout
47
- Orientation: horizontal
48
- Error Category: insufficient_extension
49
- ```
50
-
51
- ## Example Output
52
-
53
- ```
54
- Lower your hips, straight line.
55
- ```
56
-
57
- ## Usage
58
-
59
- ```python
60
- from transformers import AutoModelForCausalLM, AutoTokenizer
61
-
62
- model = AutoModelForCausalLM.from_pretrained("roli253/posefit-correction")
63
- tokenizer = AutoTokenizer.from_pretrained("roli253/posefit-correction")
64
-
65
- messages = [
66
- {"role": "system", "content": "You are a professional fitness coach. You analyze movement data and provide short, actionable correction commands."},
67
- {"role": "user", "content": "Analyze the following movement analysis and provide a correction command in English.\n\nAnalysis:\nMean Actual Angle: 133.8 degrees\nMean Target Angle: 171.6 degrees\nJoint: left_shoulder-left_hip-left_knee\nPhase: throughout\nOrientation: horizontal\nError Category: insufficient_extension"}
68
- ]
69
-
70
- text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
71
- inputs = tokenizer(text, return_tensors="pt")
72
- outputs = model.generate(**inputs, max_new_tokens=50)
73
- print(tokenizer.decode(outputs[0], skip_special_tokens=True))
74
- ```
75
-
76
- ## License
77
-
78
- Apache 2.0
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen3-4B-Instruct-2507
4
+ tags:
5
+ - fitness
6
+ - pose-correction
7
+ - exercise-coaching
8
+ - fine-tuned
9
+ language:
10
+ - en
11
+ - hu
12
+ ---
13
+
14
+ # PoseFit Correction Model
15
+
16
+ Fine-tuned Qwen3-4B model for generating real-time exercise correction commands based on pose analysis.