Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ generator = pipeline("text-generation", model="LiquidAI/LFM2-1.2B")
|
|
| 6 |
|
| 7 |
# Function to generate text from input prompt
|
| 8 |
def generate_text(prompt):
|
| 9 |
-
result = generator(prompt, max_length=
|
| 10 |
return result[0]['generated_text']
|
| 11 |
|
| 12 |
# Create a Gradio interface
|
|
|
|
| 6 |
|
| 7 |
# Function to generate text from input prompt
|
| 8 |
def generate_text(prompt):
|
| 9 |
+
result = generator(prompt, max_length=500, num_return_sequences=1)
|
| 10 |
return result[0]['generated_text']
|
| 11 |
|
| 12 |
# Create a Gradio interface
|