Detsutut commited on
Commit
979c186
1 Parent(s): 7a4dff7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ pipeline = transformers.pipeline(
12
  )
13
 
14
  # Define the function to generate text
15
- def generate_text(input_text, max_new_tokens, temperature, top_k, top_p, split_output):
16
  output = pipeline(
17
  input_text,
18
  max_new_tokens=max_new_tokens,
@@ -72,4 +72,4 @@ with gr.Blocks() as iface:
72
 
73
  # Launch the interface
74
  if __name__ == "__main__":
75
- iface.launch()
 
12
  )
13
 
14
  # Define the function to generate text
15
+ def generate_text(input_text, max_new_tokens, temperature, top_p, split_output):
16
  output = pipeline(
17
  input_text,
18
  max_new_tokens=max_new_tokens,
 
72
 
73
  # Launch the interface
74
  if __name__ == "__main__":
75
+ iface.launch()