Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ messages = [
|
|
| 24 |
|
| 25 |
@spaces.GPU
|
| 26 |
def generate_text(prompt):
|
| 27 |
-
response = pipe(prompt,
|
| 28 |
return response[0]['generated_text']
|
| 29 |
|
| 30 |
# Custom CSS for the app
|
|
@@ -72,7 +72,7 @@ with gr.Blocks(css=css) as journal_app:
|
|
| 72 |
|
| 73 |
with gr.Row():
|
| 74 |
# Output for generated text
|
| 75 |
-
output = gr.
|
| 76 |
with gr.Row():
|
| 77 |
# Generate button
|
| 78 |
generate_button = gr.Button("Generate Entry ✍️")
|
|
|
|
| 24 |
|
| 25 |
@spaces.GPU
|
| 26 |
def generate_text(prompt):
|
| 27 |
+
response = pipe(prompt, max_new_tokens=1024, num_return_sequences=1)
|
| 28 |
return response[0]['generated_text']
|
| 29 |
|
| 30 |
# Custom CSS for the app
|
|
|
|
| 72 |
|
| 73 |
with gr.Row():
|
| 74 |
# Output for generated text
|
| 75 |
+
output = gr.Markdown(label="Your AI Journal Entry ✨", lines=6)
|
| 76 |
with gr.Row():
|
| 77 |
# Generate button
|
| 78 |
generate_button = gr.Button("Generate Entry ✍️")
|