.

.gitignore

@@ -1 +1,2 @@
-secrets.py
+secrets.py
+.streamlit

app.py

@@ -1,20 +1,11 @@
 import streamlit as st
-from langchain_community.llms import Ollama
 from transformers import pipeline
-from huggingface_hub import login
-import os 
-
-
-
-# Log in with the token
-login()
-
+import os
 
 
 # Streamlit app title
 st.title("Ollama LLaMA-2 Prompt Interface")
-generator = pipeline('text-generation', model='meta-llama/Llama-2-7b-hf')
-
+generator = pipeline('text-generation', model='meta-llama/Llama-2-7b-hf', use_auth_token=os.getenv("HF_TOKEN"))
 
 user_prompt = st.text_input("Enter your prompt:", "The first man on the moon was ...")
 
@@ -24,5 +15,3 @@ if st.button("Generate Response"):
     # Display the result in the Streamlit app
     st.write("Response:")
     st.write(response)
-
-