Update app.py

app.py

@@ -1,53 +1,46 @@
 import streamlit as st
-from langchain.prompts import PromptTemplate
-from langchain.llms import CTransformers
-
-## Function To get response from LLAma 2 model
-
-def getLLamaresponse(input_text,keywords,blog_style):
-
-    ### LLama2 model
-    llm=CTransformers(model="models/llama-2-7b-chat.ggmlv3.q2_K.bin",
-                      model_type='llama',
-                      config={'max_new_tokens':250,
-                              'temperature':0.01})
-    
-    ## Prompt Template
-
-    template="""
-        Generate project idea for {blog_style} by using keywords like {keywords} for the profession of {input_text}  .
-            """
-    
-    prompt=PromptTemplate(input_variables=["blog_style","input_text",'keywords'],
-                          template=template)
-    
-    ## Generate the ressponse from the LLama 2 model
-    response=llm(prompt.format(blog_style=blog_style,input_text=input_text,keywords=keywords))
-    print(response)
-    return response
+from transformers import pipeline
 
-st.set_page_config(page_title="Generate Project Idea",
-                    page_icon='🤖',
-                    layout='centered',
-                    initial_sidebar_state='collapsed')
+# Function to get response from LLaMA 2 model
 
-st.header("Generate Project Idea 🤖")
+def getLLamaresponse(input_text, keywords, blog_style):
+    # Load the LLaMA 2 model from Hugging Face
+    model_name = "meta-llama/Llama-2-7b-chat-hf"
+    llm = pipeline('text-generation', model=model_name)
+
+    # Prompt Template
+    template = """
+        Generate project idea for {blog_style} by using keywords like {keywords} for the profession of {input_text}.
+    """
+
+    # Format the prompt
+    prompt = template.format(blog_style=blog_style, input_text=input_text, keywords=keywords)
 
-input_text=st.text_input("Enter the Topic")
+    # Generate the response from the LLaMA 2 model
+    response = llm(prompt, max_length=250, temperature=0.01)
+    return response[0]['generated_text']
 
+st.set_page_config(page_title="Generate Project Idea",
+                   page_icon='🤖',
+                   layout='centered',
+                   initial_sidebar_state='collapsed')
+
+st.header("Generate Project Idea 🤖")
 
-## creating to more columns for additonal 2 fields
+input_text = st.text_input("Enter the Topic")
 
-col1,col2=st.columns([5,5])
+# Creating two more columns for additional fields
+col1, col2 = st.columns([5, 5])
 
 with col1:
-    no_words=st.text_input('Keywords')
+    no_words = st.text_input('Keywords')
 with col2:
-    blog_style=st.selectbox('Generating project idea for',
-                            ('Researchers','Data Scientist','Software Developer','Common People', " "),index=0)
-    
-submit=st.button("Generate")
+    blog_style = st.selectbox('Generating project idea for',
+                              ('Researchers', 'Data Scientist', 'Software Developer', 'Common People', " "), index=0)
+
+submit = st.button("Generate")
 
-## Final response
+# Final response
 if submit:
-    st.write(getLLamaresponse(input_text,no_words,blog_style))
+    response = getLLamaresponse(input_text, no_words, blog_style)
+    st.write(response)