Upload 3 files

README.md

@@ -0,0 +1,12 @@
+---
+title: "[NSFW] C0ffee's Erotic Story Generator 2"
+emoji: 🍑
+colorFrom: gray
+colorTo: pink
+sdk: gradio
+sdk_version: 3.27.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,100 @@
+import gradio as gr
+
+import nltk
+import string
+from transformers import GPT2LMHeadModel, GPT2Tokenizer, GenerationConfig, set_seed
+import random
+
+nltk.download('punkt')
+
+response_length = 200
+
+sentence_detector = nltk.data.load('tokenizers/punkt/english.pickle')
+
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
+tokenizer.truncation_side = 'right'
+
+# model = GPT2LMHeadModel.from_pretrained('checkpoint-50000')
+model = GPT2LMHeadModel.from_pretrained('coffeeee/nsfw-story-generator2')
+generation_config = GenerationConfig.from_pretrained('gpt2-medium')
+generation_config.max_new_tokens = response_length
+generation_config.pad_token_id = generation_config.eos_token_id
+def generate_response(outputs, new_prompt):
+
+    story_so_far = "\n".join(outputs[:int(1024 / response_length + 1)]) if outputs else ""
+
+    set_seed(random.randint(0, 4000000000))
+    inputs = tokenizer.encode(story_so_far + "\n" + new_prompt if story_so_far else new_prompt,
+                              return_tensors='pt', truncation=True,
+                              max_length=1024 - response_length)
+
+    output = model.generate(inputs, do_sample=True, generation_config=generation_config)
+
+    response = clean_paragraph(tokenizer.batch_decode(output)[0][(len(story_so_far) + 1 if story_so_far else 0):])
+    outputs.append(response)
+    return {
+        user_outputs: outputs,
+        story: (story_so_far + "\n" if story_so_far else "") + response,
+        prompt: None
+    }
+
+def undo(outputs):
+
+    outputs = outputs[:-1] if outputs else []
+    return {
+        user_outputs: outputs,
+        story: "\n".join(outputs) if outputs else None
+    }
+
+def clean_paragraph(entry):
+    paragraphs = entry.split('\n')
+
+    for i in range(len(paragraphs)):
+        split_sentences = nltk.tokenize.sent_tokenize(paragraphs[i], language='english')
+        if i == len(paragraphs) - 1 and split_sentences[:1][-1] not in string.punctuation:
+            paragraphs[i] = " ".join(split_sentences[:-1])
+
+    return capitalize_first_char("\n".join(paragraphs))
+
+def reset():
+    return {
+        user_outputs: [],
+        story: None
+    }
+
+def capitalize_first_char(entry):
+    for i in range(len(entry)):
+        if entry[i].isalpha():
+            return entry[:i] + entry[i].upper() + entry[i + 1:]
+    return entry
+
+with gr.Blocks(theme=gr.themes.Default(text_size='lg', font=[gr.themes.GoogleFont("Bitter"), "Arial", "sans-serif"])) as demo:
+
+    placeholder_text = '''
+    Disclaimer: everything this model generates is a work of fiction.
+    Content from this model WILL generate inappropriate and potentially offensive content.
+
+    Use at your own discretion. Please respect the Huggingface code of conduct.'''
+
+    story = gr.Textbox(label="Story", interactive=False, lines=20, placeholder=placeholder_text)
+    story.style(show_copy_button=True)
+
+    user_outputs = gr.State([])
+
+    prompt = gr.Textbox(label="Prompt", placeholder="Start a new story, or continue your current one!", lines=3, max_lines=3)
+
+    with gr.Row():
+        gen_button = gr.Button('Generate')
+        undo_button = gr.Button("Undo")
+        res_button = gr.Button("Reset")
+
+    prompt.submit(generate_response, [user_outputs, prompt], [user_outputs, story, prompt], scroll_to_output=True)
+    gen_button.click(generate_response, [user_outputs, prompt], [user_outputs, story, prompt], scroll_to_output=True)
+    undo_button.click(undo, user_outputs, [user_outputs, story], scroll_to_output=True)
+    res_button.click(reset, [], [user_outputs, story], scroll_to_output=True)
+
+# for local server; comment out for deploy
+
+demo.launch(inbrowser=True, server_name='0.0.0.0')
+
+

requirements.txt

@@ -0,0 +1,78 @@
+aiofiles==23.1.0
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==4.2.2
+altgraph==0.17.3
+anyio==3.6.2
+async-timeout==4.0.2
+attrs==23.1.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+click==8.1.3
+colorama==0.4.6
+contourpy==1.0.7
+cycler==0.11.0
+entrypoints==0.4
+fastapi==0.95.1
+ffmpy==0.3.0
+filelock==3.12.0
+fonttools==4.39.3
+frozenlist==1.3.3
+fsspec==2023.4.0
+gradio==3.27.0
+gradio_client==0.1.3
+h11==0.14.0
+httpcore==0.17.0
+httpx==0.24.0
+huggingface-hub==0.14.0
+idna==3.4
+Jinja2==3.1.2
+joblib==1.2.0
+jsonschema==4.17.3
+kiwisolver==1.4.4
+linkify-it-py==2.0.0
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+matplotlib==3.7.1
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.0.4
+networkx==3.1
+nltk==3.8.1
+numpy==1.24.3
+orjson==3.8.10
+packaging==23.1
+pandas==2.0.1
+pefile==2023.2.7
+Pillow==9.5.0
+pydantic==1.10.7
+pydub==0.25.1
+pyinstaller==5.10.1
+pyinstaller-hooks-contrib==2023.2
+pyparsing==3.0.9
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3
+pywin32-ctypes==0.2.0
+PyYAML==6.0
+regex==2023.3.23
+requests==2.28.2
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.26.1
+sympy==1.11.1
+tokenizers==0.13.3
+toolz==0.12.0
+torch==2.0.0
+tqdm==4.65.0
+transformers==4.28.1
+typing_extensions==4.5.0
+tzdata==2023.3
+uc-micro-py==1.0.1
+urllib3==1.26.15
+uvicorn==0.21.1
+websockets==11.0.2
+yarl==1.9.1