Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ Columns:
|
|
52 |
Output: """
|
53 |
|
54 |
def preprocess_user_prompt(user_prompt):
|
55 |
-
generated_text = text_generator(user_prompt, max_length=
|
56 |
return generated_text
|
57 |
|
58 |
def format_prompt(description, columns):
|
|
|
52 |
Output: """
|
53 |
|
54 |
def preprocess_user_prompt(user_prompt):
|
55 |
+
generated_text = text_generator(user_prompt, max_length=60, num_return_sequences=1)[0]["generated_text"]
|
56 |
return generated_text
|
57 |
|
58 |
def format_prompt(description, columns):
|