la-zorra-negra's picture
Upload app.py with huggingface_hub
411d01b verified
Raw
History Blame Contribute Delete
843 Bytes
import streamlit as st
import ollama as ol
import time
desiredModel = "deepseek-r1:1.5b"
st.title("protoype app using deepseek-r1:1.5b")
def generate_response(questionToAsk):
response = ol.chat(model=desiredModel, messages=[
{
'role': 'user',
'content' : questionToAsk,
}
]
)
st.info(response['message']['content'])
with st.form("my_form"):
text = st.text_area(
"Enter text:",
)
st.write("Would you like to be reminded in 5 minutes?")
checkbox_val = st.checkbox("Yes")
submitted = st.form_submit_button("Submit")
if submitted:
with st.status("Reminder in 5 minutes..."):
time.sleep(300)
st.write("All done!")
generate_response(text)