Add library name to model card
Browse filesThis PR ensures the "how to use" button appears on the top right (with a Transformers code snippet), ensuring people can easier use your model.
README.md
CHANGED
@@ -1,15 +1,16 @@
|
|
1 |
---
|
2 |
-
license: apache-2.0
|
3 |
-
language:
|
4 |
-
- en
|
5 |
base_model:
|
6 |
- meta-llama/Llama-3.1-8B-Instruct
|
|
|
|
|
|
|
7 |
pipeline_tag: text-generation
|
8 |
tags:
|
9 |
- rag
|
|
|
10 |
---
|
11 |
|
12 |
-
|
13 |
<div align="center">
|
14 |
<b style="font-size: 40px;">Ext2Gen-8B-R2</b>
|
15 |
</div>
|
@@ -63,7 +64,9 @@ def prepare_sample_text(prompt):
|
|
63 |
def format_prompt_template(query, chunk_list):
|
64 |
|
65 |
chunk_list = ['[Chunk ID: '+ str(idx+1) + '] ' + chunk_text for idx, chunk_text in enumerate(chunk_list)]
|
66 |
-
chunk_list = '
|
|
|
|
|
67 |
|
68 |
prompt = '''
|
69 |
You are an expert assistant trained to extract essential sentences from document chunks and generate answers based on the extracted sentences.
|
@@ -141,6 +144,4 @@ Our evaluations demonstrate that Ext2Gen-8B-R2 significantly enhances robustness
|
|
141 |
See the results in the Figure below:
|
142 |
|
143 |

|
144 |
-
|
145 |
-
|
146 |
-
|
|
|
1 |
---
|
|
|
|
|
|
|
2 |
base_model:
|
3 |
- meta-llama/Llama-3.1-8B-Instruct
|
4 |
+
language:
|
5 |
+
- en
|
6 |
+
license: apache-2.0
|
7 |
pipeline_tag: text-generation
|
8 |
tags:
|
9 |
- rag
|
10 |
+
library_name: transformers
|
11 |
---
|
12 |
|
13 |
+
```markdown
|
14 |
<div align="center">
|
15 |
<b style="font-size: 40px;">Ext2Gen-8B-R2</b>
|
16 |
</div>
|
|
|
64 |
def format_prompt_template(query, chunk_list):
|
65 |
|
66 |
chunk_list = ['[Chunk ID: '+ str(idx+1) + '] ' + chunk_text for idx, chunk_text in enumerate(chunk_list)]
|
67 |
+
chunk_list = '
|
68 |
+
|
69 |
+
'.join(chunk_list)
|
70 |
|
71 |
prompt = '''
|
72 |
You are an expert assistant trained to extract essential sentences from document chunks and generate answers based on the extracted sentences.
|
|
|
144 |
See the results in the Figure below:
|
145 |
|
146 |

|
147 |
+
```
|
|
|
|