README / app.py
Mo-alaa's picture
Update app.py
089b91a
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from langchain.llms import HuggingFaceEndpoint
import streamlit as st
# prepare Falcon Huggingface API
llm = HuggingFaceEndpoint(
endpoint_url= f"https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
huggingfacehub_api_token='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',
task="text-generation",
model_kwargs = {
"min_length": 8192,
"max_length":8192,
"temperature":0.1,
"max_new_tokens":4000,
"num_return_sequences":1
}
)
topics_template = PromptTemplate(
input_variables = ['keyword'],
template = """
I want you to Write a comprehensive article about "{keyword}" covering the following aspects:
Introduction: Provide an engaging introduction to the topic, highlighting its relevance and significance.
History and Background: Explain the historical context and background of {keyword}.
Key Concepts and Terminology: Define important terms and concepts related to {keyword}.
Current State of {keyword}: Discuss the current trends, developments, and challenges in the field of {keyword}.
Use Cases and Applications: Explore practical applications and use cases of {keyword} in various industries.
Benefits and Drawbacks: Highlight the advantages and disadvantages of {keyword}.
Future Outlook: Predict the future trends and potential advancements in {keyword}.
Conclusion: Summarize the key points and reiterate the importance of {keyword}.
Ensure that the article is well-structured, informative, and at least 1500 words long. Use SEO best practices for content optimization.
""")
st.title("Blog Writer")
keyword = st.text_input("Input the keyword you wish to write about")
topic_writing_chain = LLMChain(llm= llm, prompt=topics_template, verbose = True)
with st.spinner('Wait for it...'):
if keyword:
topics = topic_writing_chain(keyword)
st.write(topics)
st.success('')