cakemus commited on
Commit
dee82dd
·
1 Parent(s): 36dec11
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Load a large model (e.g., text generation)
5
- model = pipeline("text-generation", model="gpt-3")
6
 
7
  def generate_text(prompt):
8
- return model(prompt, max_length=50)
9
 
10
- # Custom interface with sliders and descriptions
11
  interface = gr.Interface(
12
  fn=generate_text,
13
  inputs=gr.Textbox(label="Enter your prompt here"),
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Load an available model for text generation
5
+ model = pipeline("text-generation", model="gpt2")
6
 
7
  def generate_text(prompt):
8
+ return model(prompt, max_length=50)[0]["generated_text"]
9
 
10
+ # Custom interface with a textbox and description
11
  interface = gr.Interface(
12
  fn=generate_text,
13
  inputs=gr.Textbox(label="Enter your prompt here"),