Vikram Thangaraj commited on
Commit
5d2f5b2
·
1 Parent(s): c6927d3
Files changed (1) hide show
  1. app.py +1 -1
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=50, num_return_sequences=1)
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