Spaces:
Build error
Build error
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")
|