K. K commited on
Commit
5846235
1 Parent(s): 33613fe

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline, set_seed
3
+
4
+ # Create text generation pipeline with GPT-2
5
+ generator = pipeline('text-generation', model='gpt2')
6
+
7
+ def generate_text(prompt, temperature, max_length, instruction):
8
+ set_seed(42)
9
+ result = generator(prompt, max_length=max_length, num_return_sequences=1, temperature=temperature)
10
+ return result[0]['generated_text']
11
+
12
+ # Define Gradio interface
13
+ iface = gr.Interface(
14
+ fn=generate_text,
15
+ inputs=[gr.inputs.Textbox(lines=3, label="Your Message"),
16
+ gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.5, label="Temperature"),
17
+ gr.inputs.Slider(minimum=10, maximum=200, default=100, label="Max Length"),
18
+ gr.inputs.Textbox(default="This is a chat with AI.", lines=2, label="AI Instruction")],
19
+ outputs=gr.outputs.Textbox(label="AI Response"),
20
+ layout="vertical",
21
+ title="GPT-2 Chat",
22
+ description="This is a chat interface with OpenAI's GPT-2 model. You can adjust the temperature and the max length of the response. The temperature controls the randomness of the model's output. Higher values (closer to 1) will make the output more random, while lower values will make it more deterministic. The max length controls the length of the model's response.",
23
+ )
24
+
25
+ iface.launch()