ksh-nyp commited on
Commit
6cba48c
1 Parent(s): 996946f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ pipe = pipeline("text-generation", model=model_name)
7
 
8
  def generate_text(prompt):
9
  # Generate text based on the input prompt
10
- results = pipe(prompt, max_length=1024) # Adjust max_length as needed
11
  return results[0]['generated_text']
12
 
13
  # Create the Gradio interface
 
7
 
8
  def generate_text(prompt):
9
  # Generate text based on the input prompt
10
+ results = pipe(prompt, max_length=1024)
11
  return results[0]['generated_text']
12
 
13
  # Create the Gradio interface