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 |
|