Update README.md
Browse files
README.md
CHANGED
|
@@ -22,9 +22,9 @@ Fine-tuned model of Arabic poetry dataset based on aragpt2-medium.
|
|
| 22 |
|
| 23 |
#### How to use
|
| 24 |
|
| 25 |
-
Try this [HF Space](https://huggingface.co/spaces/akhooli/poetry).
|
| 26 |
-
From script:
|
| 27 |
-
|
| 28 |
from transformers import pipeline
|
| 29 |
pipe = pipeline('text-generation', framework='pt', device=-1, model='akhooli/ap2023', tokenizer='akhooli/ap2023')
|
| 30 |
gen = pipe(prompt, max_length=96, temperature = 0.95,repetition_penalty=1.05,
|
|
@@ -34,7 +34,7 @@ poetry =""
|
|
| 34 |
for line in gen.split('.')[:-1]:
|
| 35 |
poetry += line
|
| 36 |
print(poetry)
|
| 37 |
-
|
| 38 |
|
| 39 |
#### Limitations and bias
|
| 40 |
|
|
|
|
| 22 |
|
| 23 |
#### How to use
|
| 24 |
|
| 25 |
+
Try this [HF Space](https://huggingface.co/spaces/akhooli/poetry).
|
| 26 |
+
From script:
|
| 27 |
+
```
|
| 28 |
from transformers import pipeline
|
| 29 |
pipe = pipeline('text-generation', framework='pt', device=-1, model='akhooli/ap2023', tokenizer='akhooli/ap2023')
|
| 30 |
gen = pipe(prompt, max_length=96, temperature = 0.95,repetition_penalty=1.05,
|
|
|
|
| 34 |
for line in gen.split('.')[:-1]:
|
| 35 |
poetry += line
|
| 36 |
print(poetry)
|
| 37 |
+
```
|
| 38 |
|
| 39 |
#### Limitations and bias
|
| 40 |
|