MarkProMaster229 commited on
Commit
800a0d5
·
verified ·
1 Parent(s): 1ec18d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ model = pipeline("text-generation", model="MarkProMaster229/arch")
5
 
6
 
7
  def generate_text(prompt):
8
- return model(prompt, max_length=100)[0]['generated_text']
9
 
10
  demo = gr.Interface(
11
  fn=generate_text,
 
5
 
6
 
7
  def generate_text(prompt):
8
+ return model(prompt, max_length=50)[0]['generated_text']
9
 
10
  demo = gr.Interface(
11
  fn=generate_text,