aaa / app.py
lineee's picture
Upload 2 files
e0c43de verified
from langchain_community.llms import Ollama
import streamlit as st
llm = Ollama(model="llama3:70b")
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|>']))