drfahlan's picture
Update app.py
3223286 verified
raw history blame
No virus
633 Bytes
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)