Hugging Face's logo

Spaces:
ssalb
/
story_generator
Paused

App Files Community
story_generator / README.md
ssalb
Update space with latest code and dependencies on Mon Jan 13 19:33:51 UTC 2025
139df7a
preview code
|
raw
history blame contribute delete
1.29 kB

A newer version of the Gradio SDK is available: 5.13.1

Upgrade
metadata
title: Story Generator
emoji: 📚
colorFrom: blue
python_version: 3.10.13
colorTo: pink
sdk: gradio
sdk_version: 5.9.1
app_file: app.py
pinned: false
preload_from_hub:
  - HuggingFaceTB/SmolLM2-135M-Instruct
  - answerdotai/ModernBERT-base
  - facebook/bart-large-mnli
license: mit

Project Overview

This Story Generator leverages natural language processing models to generate coherent and engaging stories. By utilizing models such as SmolLMv2, BERT, and BART, this project aims to provide users with a tool to create narratives based on given prompts. The application is built using Gradio for an interactive user interface, making it easy to input prompts and receive generated stories in real-time.

The main purpose of this project is to explore the idea of beam search for selecting stories with high coherence, fluency, and genre alignment scores in a process-based reward model (PRM) fashion. This ensures that the generated stories are not only creative but also maintain a logical flow and adhere to the specified genre.

Note that the final implementation is not strictly beam search and was modified to allow more diversity (creativity) inspired by the DVTS method in this blog post.