NadaAljohani commited on
Commit
3039869
1 Parent(s): 2d3db97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ demo = gr.Interface(
31
  fn=generate_story,
32
  inputs=[
33
  gr.Textbox(label="Enter Story Title", placeholder="Type a title here..."), # Title input
34
- gr.Dropdown(choices=['gpt2', 'gpt2-large', 'EleutherAI/gpt-neo-2.7B','maldv/badger-writer-llama-3-8b'], value='gpt2', label="Select Model") # Model selection
35
  ],
36
  outputs="text",
37
  title="AI Story Generator",
 
31
  fn=generate_story,
32
  inputs=[
33
  gr.Textbox(label="Enter Story Title", placeholder="Type a title here..."), # Title input
34
+ gr.Dropdown(choices=['gpt2', 'gpt2-large', 'EleutherAI/gpt-neo-2.7B','maldv/badger-writer-llama-3-8b', 'GOAT-AI/GOAT-70B-Storytelling'], value='gpt2', label="Select Model") # Model selection
35
  ],
36
  outputs="text",
37
  title="AI Story Generator",