Arpit1234 commited on
Commit
f82ce2f
·
verified ·
1 Parent(s): 1410c0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -19
app.py CHANGED
@@ -1,22 +1,6 @@
1
  import gradio as gr
2
- import transformers
3
- from transformers import pipeline
4
 
5
- # Load GPT-2 model
6
- generator = pipeline('text-generation', model='gpt2')
7
 
8
- # Define function to generate text using GPT-2
9
- def generate_text(prompt):
10
- return generator(prompt, max_length=50, do_sample=True)[0]['generated_text']
11
-
12
- # Create Gradio interface
13
- interface = gr.Interface(
14
- fn=generate_text,
15
- inputs="text",
16
- outputs="text",
17
- title="Text Generator AI",
18
- description="Generate text using GPT-2 from Hugging Face"
19
- )
20
-
21
- # Launch the interface
22
- interface.launch()
 
1
  import gradio as gr
 
 
2
 
3
+ def say_hello():
4
+ return "Hello"
5
 
6
+ gr.Interface(fn=say_hello, title="Hello App").launch()