Spaces:
Build error
Build error
File size: 342 Bytes
315e6ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from langchain_community.llms import Ollama
import streamlit as st
llm = Ollama(model="llama3:8b")
st.title("Chatbot using Llama3")
prompt = st.text_area("Enter your prompt:")
if st.button("Generate"):
if prompt:
with st.spinner("Generating response..."):
st.write_stream(llm.stream(prompt, stop=['<|eot_id|>']))
|