File size: 1,286 Bytes
3b5aca7
 
bfb4432
 
 
 
3b5aca7
 
 
 
bfb4432
139df7a
d2e0b39
bfb4432
3b5aca7
 
 
bfb4432
 
139df7a
bfb4432
139df7a
bfb4432
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
---
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](https://huggingface.co/spaces/HuggingFaceH4/blogpost-scaling-test-time-compute).