akhooli commited on
Commit
d825853
·
1 Parent(s): 08e4870

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -23,6 +23,18 @@ Fine-tuned model of Arabic poetry dataset based on aragpt2-medium.
23
  #### How to use
24
 
25
  Try this [HF Space](https://huggingface.co/spaces/akhooli/poetry).
 
 
 
 
 
 
 
 
 
 
 
 
26
 
27
  #### Limitations and bias
28
 
 
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,
31
+ num_beams=3, num_return_sequences=2, do_sample = True,
32
+ top_p = 1.0, top_k = 50, return_full_text=True)[0]["generated_text"]
33
+ poetry =""
34
+ for line in gen.split('.')[:-1]:
35
+ poetry += line
36
+ print(poetry)
37
+ `
38
 
39
  #### Limitations and bias
40