Spaces:
Sleeping
Sleeping
Vikram Thangaraj
commited on
Commit
·
5d2f5b2
1
Parent(s):
c6927d3
Update
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
|
10 |
def generate_kavithai(prompt):
|
11 |
if not prompt.strip():
|
12 |
return "தயவுசெய்து ஒரு வரியை உள்ளிடவும்..."
|
13 |
-
outputs = generator(prompt, max_length=
|
14 |
return outputs[0]["generated_text"]
|
15 |
|
16 |
# Gradio UI setup
|
|
|
10 |
def generate_kavithai(prompt):
|
11 |
if not prompt.strip():
|
12 |
return "தயவுசெய்து ஒரு வரியை உள்ளிடவும்..."
|
13 |
+
outputs = generator(prompt, max_length=1000, num_return_sequences=1)
|
14 |
return outputs[0]["generated_text"]
|
15 |
|
16 |
# Gradio UI setup
|