krishna34-aimlcurious commited on
Commit
64f5883
·
verified ·
1 Parent(s): 33479ba

Upload 2 files

Browse files
Files changed (2) hide show
  1. APP.py +34 -0
  2. requirements.txt +3 -1
APP.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # GPT-2 Text Generation App with Gradio
2
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
+ import gradio as gr
4
+
5
+ # Load GPT-2 model and tokenizer
6
+ model = GPT2LMHeadModel.from_pretrained("gpt2")
7
+ tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
8
+
9
+ # Define a function to generate text
10
+ def generate_text(prompt, max_length=50, temperature=1.0, top_p=0.95):
11
+ inputs = tokenizer(prompt, return_tensors="pt")
12
+ outputs = model.generate(
13
+ inputs.input_ids,
14
+ max_length=max_length,
15
+ do_sample=True,
16
+ temperature=temperature,
17
+ top_p=top_p,
18
+ pad_token_id=tokenizer.eos_token_id
19
+ )
20
+ return tokenizer.decode(outputs[0], skip_special_tokens=True)
21
+
22
+ # Build the Gradio app interface
23
+ gr.Interface(
24
+ fn=generate_text,
25
+ inputs=[
26
+ gr.Textbox(lines=2, label="Enter your prompt here"),
27
+ gr.Slider(20, 100, value=50, step=10, label="Max Length"),
28
+ gr.Slider(0.1, 2.0, value=1.0, step=0.1, label="Temperature"),
29
+ gr.Slider(0.5, 1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
30
+ ],
31
+ outputs="textbox",
32
+ title="🔮 GPT-2 Text Generator",
33
+ description="Write a prompt and let GPT-2 continue the text. Adjust the sliders for more creative control!"
34
+ ).launch()
requirements.txt CHANGED
@@ -1 +1,3 @@
1
- huggingface_hub==0.25.2
 
 
 
1
+ transformers
2
+ torch
3
+ gradio