srinikesh1432 commited on
Commit
d378ff9
·
verified ·
1 Parent(s): c7992d8

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +32 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # Load text generation pipeline
5
+ generator = pipeline("text-generation", model="gpt2")
6
+
7
+ def generate_text(prompt, max_length=100, temperature=0.7):
8
+ result = generator(
9
+ prompt,
10
+ max_length=max_length,
11
+ num_return_sequences=1,
12
+ temperature=temperature,
13
+ pad_token_id=50256 # for GPT2 to avoid warnings
14
+ )
15
+ return result[0]['generated_text']
16
+
17
+ with gr.Blocks() as demo:
18
+ gr.Markdown("## Text Generation App (GPT-2 Demo)")
19
+
20
+ prompt = gr.Textbox(
21
+ lines=3,
22
+ placeholder="Enter a starting sentence...",
23
+ label="Prompt"
24
+ )
25
+ length = gr.Slider(50, 300, value=100, step=10, label="Max Length")
26
+ temp = gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Creativity (Temperature)")
27
+ output = gr.Textbox(lines=10, label="Generated Text")
28
+
29
+ btn = gr.Button("Generate")
30
+ btn.click(fn=generate_text, inputs=[prompt, length, temp], outputs=output)
31
+
32
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio>=3.30
2
+ transformers>=4.40
3
+ torch>=1.13