Update app.py

app.py

@@ -43,16 +43,19 @@ def main():
     word_count = st.sidebar.slider("Number of Words", min_value=50, max_value=1000, value=200, step=50)
 
     if st.sidebar.button("Generate Blog"):
-        model_id = "google/gemma-1.1-7b-it"
-        tokenizer = AutoTokenizer.from_pretrained(model_id,token =HF_TOKEN )
-        model = AutoModelForCausalLM.from_pretrained(model_id,token =HF_TOKEN )
-        pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)
-        hf = HuggingFacePipeline(pipeline=pipe)
-        chain = LLMChain(llm=hf,prompt=prompt,verbose=True)
-        aa = chain.invoke({"topic": topic,"words":word_count,"role":role})
+        repo_id = "google/gemma-1.1-7b-it"
+
+        llm = HuggingFaceEndpoint(
+            repo_id=repo_id, max_length=128, temperature=0.5, huggingfacehub_api_token=HF_TOKEN 
+        )
+        # llm_chain = LLMChain(prompt=prompt, llm=llm)
+        # print(llm_chain.run(question))
+        aa = llm_chain.run({"topic": topic,"words":word_count,"role":role})
+        st.write(aa)
+        
     
-    st.write(aa)
-    st.write("Will Come here")
+    # st.write(aa)
+    # st.write("Will Come here")
 
 if __name__ == "__main__":
     main()