Spaces:
Sleeping
Sleeping
updated the max line
Browse files
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=
|
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!
|
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 |
|