yasserrmd commited on
Commit
88688de
·
verified ·
1 Parent(s): e2046ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ messages = [
24
 
25
  @spaces.GPU
26
  def generate_text(prompt):
27
- response = pipe(prompt, max_length=100, num_return_sequences=1)
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.Textbox(label="Your AI Journal Entry ✨", lines=6)
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 ✍️")