File size: 581 Bytes
0af1295 c926cd5 0af1295 c926cd5 0af1295 c926cd5 0af1295 c926cd5 0af1295 c926cd5 0af1295 c926cd5 0af1295 c926cd5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import streamlit as st
from langchain_community.llms import Ollama
# Initialize the language model
llm = Ollama(model="tinyllama")
# Streamlit UI elements
st.title("Language Model Invocation")
st.write("Enter a prompt to get a response from the language model.")
# Text input for prompt
prompt = st.text_input("Enter a prompt:")
# Button to invoke the model
if st.button("Submit"):
if prompt:
# Generate the response
response = llm.invoke(prompt)
st.write("Response:")
st.write(response)
else:
st.write("Please enter a prompt.")
|