Duplicate from aaaaaabbbbbbbdddddddduuuuulllll/poetry

Co-authored-by: Abdullah almaedhgy <[email protected]>

.gitattributes

@@ -0,0 +1,33 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Arabic Poetry Generator
+emoji: 🐠
+colorFrom: blue
+colorTo: red
+sdk: gradio
+sdk_version: 3.6
+app_file: app.py
+license: cc-by-nc-4.0
+duplicated_from: aaaaaabbbbbbbdddddddduuuuulllll/poetry
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,53 @@
+import gc 
+import gradio as gr
+from transformers import pipeline, set_seed
+
+pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
+#gc.collect()
+samples = [['أنت'
+           ,1.0, 50, 1.0, 1.0, 114],['هل غادر'
+            ,1.0, 50, 1.0, 1.0, 114 ],['ألا ليت'
+              ,1.0, 50, 1.0, 1.0, 114 ],['يا قدس'
+                 ,1.0, 50, 1.0, 1.0, 114],['عيد بأية حال'
+                   ,1.0, 50, 1.0, 1.0, 114],['لكل شيء إذا ما'
+                    ,1.0, 50, 1.0, 1.0, 114 ],['.'
+                       ,1.0, 50, 1.0, 1.0, 114]]
+                     
+notes = """
+- Enter a short prompt or select (click) one of the examples and click SEND 
+- Adjust parameters (temperture, top k, top p and penalty) through the slider (keep close to default values).
+- For the same seed (randomness), the same output is regenerated if other parameters are fixed. Seed should be 0 or more (not empty) 
+- Clear and enter new prompt or select another example and SEND to regenerate
+- The '.' means start a new line from no prompt (your prompt need not be long) 
+- Be patient: this runs on CPU (free tier)
+- Feedback (Twitter): @akhooli (https://twitter.com/akhooli/status/1611025232201977859) 
+- Note/Disclaimer: may generate unaccepted or inappropriate content. Use at your own risk.
+"""
+def sayPoetry(prompt, temp=1.0, topk = 50, topp = 1.0, penalty=1.0, seed=114):
+    if not int(seed) >= 0: seed=114
+    set_seed(seed)
+    gen = pipe(prompt, max_length=96, do_sample=True, temperature=temp, top_k=topk, top_p=topp, repetition_penalty=penalty,
+              min_length = 64, no_repeat_ngram_size = 3, return_full_text=True,
+               num_beams=5, num_return_sequences=1)[0]["generated_text"]
+    poetry =""
+    for line in gen.split('.')[:-1]:
+        poetry += line #+ "\n"
+    return poetry
+poetry = gr.Interface(fn=sayPoetry, 
+                    inputs=[
+                        gr.Textbox(label="Enter short prompt or select from examples:"), 
+                        gr.Slider(0.70, 1.2, step=0.01,value=1.0, label='control  temperature'),
+                        gr.Slider(25, 100, step=1,value=50, label='control  top k'),
+                        gr.Slider(0.80, 1.0, step=0.01,value=1.0, label='control  top p'),
+                        gr.Slider(0.90, 1.50, step=0.01,value=1.0, label='control  penalty'),
+                        gr.Number(value=139750, precision=0, label='Seed'),
+                    ], 
+                outputs=[gr.Textbox(label="Generated Poetry:")], 
+                    
+                allow_flagging='never',
+                title='Arabic Poetry Generation Demo (updated Jan. 2023)',
+                description = "A simple demo of AI generated poetry based on 1M poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
+                examples=samples,
+                cache_examples=False,
+                article = notes)
+poetry.launch()         

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+gradio #==3.0.2
+torch