test-B / app.py
avreymi's picture
remove import torch
32d5467
import streamlit as st
from streamlit import session_state as state
import pipline
if "app" not in state:
state.app = "model"
state.out = ""
st.title("Streamlit using Huggingface Transformers and langchain")
in_area = st.container()
out_area = st.container()
def __run_pipline():
out_area.markdown(":green[Running pipline]")
out_area.text(pipline.chain_TI(state.input_text))
in_area.text_area("input_text", key="input_text")
in_area.button("run", on_click=__run_pipline)