Pooja P commited on
Commit
cb07ab0
·
1 Parent(s): e36c65e

used transformer instead of openai

Browse files
Files changed (1) hide show
  1. app.py +20 -2
app.py CHANGED
@@ -1,6 +1,24 @@
 
 
 
 
 
 
 
 
1
  from transformers import pipeline
 
2
 
 
3
  generator = pipeline("text-generation", model="gpt2")
4
 
5
- def generate_blog(topic):
6
- return generator(f"Write a blog on: {topic}", max_length=200)[0]["generated_text"]
 
 
 
 
 
 
 
 
 
1
+ # from transformers import pipeline
2
+
3
+ # generator = pipeline("text-generation", model="gpt2")
4
+
5
+ # def generate_blog(topic):
6
+ # return generator(f"Write a blog on: {topic}", max_length=200)[0]["generated_text"]
7
+
8
+
9
  from transformers import pipeline
10
+ import gradio as gr
11
 
12
+ # Load the model
13
  generator = pipeline("text-generation", model="gpt2")
14
 
15
+ # Define your function
16
+ def generate_text(prompt):
17
+ result = generator(prompt, max_length=100, num_return_sequences=1)
18
+ return result[0]["generated_text"]
19
+
20
+ # Create Gradio interface
21
+ iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Text Generator with GPT-2")
22
+
23
+ # ✅ This line is required to actually launch the app on Hugging Face Spaces
24
+ iface.launch()