Faisal-Data commited on
Commit
b57ee92
·
verified ·
1 Parent(s): 07affc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -1,7 +1,13 @@
1
- import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- demo.launch()
 
 
 
 
 
 
 
1
+ from transformers import pipeline, set_seed
2
 
3
+ # Load the GPT-2 medium model
4
+ generator = pipeline('text-generation', model='gpt2-medium')
5
 
6
+ # Set a seed for reproducibility (optional)
7
+ set_seed(42)
8
+
9
+ # Generate text
10
+ prompt = "GPT-2 is a model developed by OpenAI."
11
+ generated_text = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
12
+
13
+ print(generated_text)