prithivMLmods commited on
Commit
25caed5
Β·
verified Β·
1 Parent(s): 6ac13b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def generate_and_save(prompt, history, filename="output", file_format="pdf", sys
80
  for output in generate(prompt, history, system_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
81
  generated_text = output
82
  saved_file = save_file(generated_text, filename, file_format)
83
- return generated_text, saved_file
84
 
85
  # Create Gradio interface
86
  demo = gr.Interface(
@@ -93,6 +93,7 @@ demo = gr.Interface(
93
  ],
94
  outputs=[
95
  gr.Textbox(label="Generated Text"),
 
96
  gr.File(label="Download File")
97
  ],
98
  css=css,
 
80
  for output in generate(prompt, history, system_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
81
  generated_text = output
82
  saved_file = save_file(generated_text, filename, file_format)
83
+ return generated_text, history + [(prompt, generated_text)], saved_file
84
 
85
  # Create Gradio interface
86
  demo = gr.Interface(
 
93
  ],
94
  outputs=[
95
  gr.Textbox(label="Generated Text"),
96
+ gr.State(value=[]), # history
97
  gr.File(label="Download File")
98
  ],
99
  css=css,