import streamlit as st
from transformers import AutoTokenizer, AutoModel

# Load model and tokenizer
tokenizer = AutoTokenizer.from_pretrained("medicalai/ClinicalBERT")
model = AutoModel.from_pretrained("medicalai/ClinicalBERT")

# Define app layout
st.title("ClinicalBERT Inference App")
input_text = st.text_input("Enter clinical text:")

if input_text:
    # Preprocess input
    inputs = tokenizer(input_text, return_tensors="pt")

    # Perform inference
    with st.spinner("Processing..."):
        outputs = model(**inputs)

    # Display results (adjust based on model output format)
    st.write("Model outputs:", outputs)