Update app.py
Browse files
app.py
CHANGED
|
@@ -21,12 +21,12 @@ After filling in these details, click the "Submit" button, and a short story wil
|
|
| 21 |
"""
|
| 22 |
|
| 23 |
inputs = [
|
| 24 |
-
gr.
|
| 25 |
-
gr.
|
| 26 |
-
gr.
|
| 27 |
]
|
| 28 |
|
| 29 |
-
outputs = gr.
|
| 30 |
|
| 31 |
# Load the tokenizer and model
|
| 32 |
model_path = "nvidia/Minitron-8B-Base"
|
|
|
|
| 21 |
"""
|
| 22 |
|
| 23 |
inputs = [
|
| 24 |
+
gr.Textbox(label="Main Character", placeholder="e.g. a brave knight"),
|
| 25 |
+
gr.Textbox(label="Setting", placeholder="e.g. in an enchanted forest"),
|
| 26 |
+
gr.Textbox(label="Plot Twist", placeholder="e.g. discovers a hidden treasure")
|
| 27 |
]
|
| 28 |
|
| 29 |
+
outputs = gr.Textbox(label="Generated Story")
|
| 30 |
|
| 31 |
# Load the tokenizer and model
|
| 32 |
model_path = "nvidia/Minitron-8B-Base"
|