Fix pipeline tag, correct metadata typo, and add paper link

#1
by nielsr HF staff - opened
Files changed (1) hide show
  1. README.md +9 -11
README.md CHANGED
@@ -1,18 +1,17 @@
1
  ---
2
  base_model:
3
  - Qwen/Qwen2-VL-2B-Instruct
 
 
 
 
4
  library_name: transformers
 
5
  model_name: HazardNet-unsloth-v0.4
 
6
  tags:
7
  - trl
8
  - sft
9
- licence: license
10
- license: apache-2.0
11
- datasets:
12
- - Tami3/HazardQA
13
- language:
14
- - en
15
- pipeline_tag: visual-question-answering
16
  ---
17
 
18
  # Model Card for HazardNet-unsloth-v0.4
@@ -20,6 +19,8 @@ pipeline_tag: visual-question-answering
20
  This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
21
  It has been trained using [TRL](https://github.com/huggingface/trl).
22
 
 
 
23
  ## Quick start
24
 
25
  ```python
@@ -30,7 +31,7 @@ from io import BytesIO
30
 
31
  # Initialize the Visual Question Answering pipeline with HazardNet
32
  hazard_vqa = pipeline(
33
- "visual-question-answering",
34
  model="Tami3/HazardNet"
35
  )
36
 
@@ -82,7 +83,6 @@ except Exception as e:
82
 
83
 
84
 
85
-
86
  This model was trained with SFT.
87
 
88
  ### Framework versions
@@ -95,8 +95,6 @@ This model was trained with SFT.
95
 
96
  ## Citations
97
 
98
-
99
-
100
  Cite TRL as:
101
 
102
  ```bibtex
 
1
  ---
2
  base_model:
3
  - Qwen/Qwen2-VL-2B-Instruct
4
+ datasets:
5
+ - Tami3/HazardQA
6
+ language:
7
+ - en
8
  library_name: transformers
9
+ license: apache-2.0
10
  model_name: HazardNet-unsloth-v0.4
11
+ pipeline_tag: image-text-to-text
12
  tags:
13
  - trl
14
  - sft
 
 
 
 
 
 
 
15
  ---
16
 
17
  # Model Card for HazardNet-unsloth-v0.4
 
19
  This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
20
  It has been trained using [TRL](https://github.com/huggingface/trl).
21
 
22
+ The model was presented in the paper [](https://hf.co/papers/2502.20572).
23
+
24
  ## Quick start
25
 
26
  ```python
 
31
 
32
  # Initialize the Visual Question Answering pipeline with HazardNet
33
  hazard_vqa = pipeline(
34
+ "image-text-to-text",
35
  model="Tami3/HazardNet"
36
  )
37
 
 
83
 
84
 
85
 
 
86
  This model was trained with SFT.
87
 
88
  ### Framework versions
 
95
 
96
  ## Citations
97
 
 
 
98
  Cite TRL as:
99
 
100
  ```bibtex