umairrrkhan commited on
Commit
260a648
·
verified ·
1 Parent(s): 1f8430f

updated the max line

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ def generate_text(prompt):
34
  logger.info("Generating text...")
35
  outputs = model.generate(
36
  inputs['input_ids'],
37
- max_length=190,
38
  attention_mask=inputs['attention_mask'],
39
  do_sample=True,
40
  temperature=0.7,
@@ -53,7 +53,7 @@ iface = gr.Interface(
53
  inputs="text",
54
  outputs="text",
55
  title="AI Text Generation Chatbot",
56
- description="Lowkey curious? Type a prompt and see what Its generate! (190 character limit)",
57
  examples=["Tell me a story about a robot.", "Write a poem about the moon."]
58
  )
59
 
 
34
  logger.info("Generating text...")
35
  outputs = model.generate(
36
  inputs['input_ids'],
37
+ max_length=50,
38
  attention_mask=inputs['attention_mask'],
39
  do_sample=True,
40
  temperature=0.7,
 
53
  inputs="text",
54
  outputs="text",
55
  title="AI Text Generation Chatbot",
56
+ description="Lowkey curious? Type a prompt and see what Its generate!",
57
  examples=["Tell me a story about a robot.", "Write a poem about the moon."]
58
  )
59