tamatwi commited on
Commit
0dca909
·
verified ·
1 Parent(s): 0621419

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def generate_text(prompt):
9
  return result[0]['generated_text']
10
 
11
  # Define a function to generate text based on user input
12
- # @spaces.GPU
13
  def generate_text(prompt):
14
  result = pipe(prompt, max_length=50, num_return_sequences=1)
15
  return result[0]['generated_text']
 
9
  return result[0]['generated_text']
10
 
11
  # Define a function to generate text based on user input
12
+ @spaces.GPU
13
  def generate_text(prompt):
14
  result = pipe(prompt, max_length=50, num_return_sequences=1)
15
  return result[0]['generated_text']