cafierom/InflectionLM
0
1## Project Overview2This repo contains code generating text using GEMMA 4 via the transformers library3 4### Inflections5The goal of this work is to use GEMMA 4 to generate 5 beams in response to a prompt. Each beam is parsed and can be displayed along with it's total 'score.' Each beam can also be displayed with a score-per-token.6 7#### What needs work8- create a second version of the inflections.py file with all class methods as separate functions. they should each return needed variables needed to call the next function,9- modify the app.py so that the input elemnet takes the prompt and the output displays the first beam, with any token with score < 0.6 is highlighted in red.10 11## Tech Stack12- Language: Python13- packages:14 * transformers15 * torch16 17## Project Structure18```19llm_inflections/20 inflections.py # inflection class21 app.py # gradio gui22 23```24 25## Other information26- assume any API keys can be obtained as environment variables.