yxmauw commited on
Commit
6e7b9cd
·
verified ·
1 Parent(s): fb9a8c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,9 +4,8 @@ from gpt4all import GPT4All
4
  model = GPT4All("orca-mini-3b-gguf2-q4_0.gguf")
5
 
6
  def generate_text(input_text):
7
- output = model.generate(input_text, max_tokens=50)
8
- generated_text = output["text"]
9
- return generated_text
10
 
11
  input_text = gr.Textbox(lines=5, label="Input Text")
12
  output_text = gr.Textbox(lines=5, label="Generated Text")
 
4
  model = GPT4All("orca-mini-3b-gguf2-q4_0.gguf")
5
 
6
  def generate_text(input_text):
7
+ output = model.generate(input_text, max_tokens=100)
8
+ return output
 
9
 
10
  input_text = gr.Textbox(lines=5, label="Input Text")
11
  output_text = gr.Textbox(lines=5, label="Generated Text")