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"
|