Spaces:

PoojaPoo
/

blog-writer-ai

Sleeping

Pooja P commited on Apr 10

Commit

dffd6b1

1 Parent(s): 225a7e5

used transformer instead of openai

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@
 from transformers import pipeline
 import gradio as gr
-generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
 def clean_topic(topic):
     topic = topic.lower()

 from transformers import pipeline
 import gradio as gr
+generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
 def clean_topic(topic):
     topic = topic.lower()