sailormars18 commited on
Commit
1232cab
·
1 Parent(s): 022bc31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import subprocess
2
  subprocess.run(["pip", "install","gradio","torch","transformers"])
3
  import re
4
- import gradio as gr
5
  import torch
6
  import transformers
7
 
@@ -64,14 +63,17 @@ def generate_text(prompt, length=100, theme=None, **kwargs):
64
  return generated_text
65
 
66
  # Define a Gradio interface for the generate_text function
 
 
 
 
 
 
 
67
  iface = gr.Interface(
68
  fn=generate_text,
69
- inputs=[
70
- "text",
71
- gr.inputs.Slider(minimum=10, maximum=100, default=50, label='Length of text'),
72
- gr.inputs.Textbox(default='Food', label='Theme')
73
- ],
74
- outputs=[gr.outputs.Textbox(label='Generated Text')],
75
  title='Yelp Review Generator',
76
  description='Generate a Yelp review based on a prompt, length of text, and theme.',
77
  examples=[
 
1
  import subprocess
2
  subprocess.run(["pip", "install","gradio","torch","transformers"])
3
  import re
 
4
  import torch
5
  import transformers
6
 
 
63
  return generated_text
64
 
65
  # Define a Gradio interface for the generate_text function
66
+ inputs = [
67
+ gr.inputs.Textbox(type="text", label="Text"),
68
+ gr.inputs.Slider(minimum=10, maximum=100, default=50, label="Length of text"),
69
+ gr.inputs.Textbox(type="text", label="Theme", default="Food")
70
+ ]
71
+ outputs = gr.outputs.Textbox(label="Generated Text")
72
+
73
  iface = gr.Interface(
74
  fn=generate_text,
75
+ inputs=inputs,
76
+ outputs=outputs,
 
 
 
 
77
  title='Yelp Review Generator',
78
  description='Generate a Yelp review based on a prompt, length of text, and theme.',
79
  examples=[