Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
4 |
# Set up the LLaMA 3.2 model (using Hugging Face pipeline)
|
5 |
@st.cache_resource
|
6 |
def load_llama_model():
|
7 |
-
return pipeline('text-generation', model='meta-llama/Llama-2-7b-hf')
|
8 |
|
9 |
# Load the model
|
10 |
llama_generator = load_llama_model()
|
@@ -40,4 +40,4 @@ length = st.sidebar.slider('Max Length', 50, 200, 100)
|
|
40 |
|
41 |
# Footer
|
42 |
st.sidebar.markdown("---")
|
43 |
-
st.sidebar.markdown("Built with 🧠 by Hruday & Ollama")
|
|
|
4 |
# Set up the LLaMA 3.2 model (using Hugging Face pipeline)
|
5 |
@st.cache_resource
|
6 |
def load_llama_model():
|
7 |
+
return pipeline('text-generation', model='meta-llama/Llama-3.2-7b-hf')
|
8 |
|
9 |
# Load the model
|
10 |
llama_generator = load_llama_model()
|
|
|
40 |
|
41 |
# Footer
|
42 |
st.sidebar.markdown("---")
|
43 |
+
st.sidebar.markdown("Built with 🧠 by Hruday & Ollama")
|