File size: 494 Bytes
6050982 4d2d776 6050982 4d2d776 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import streamlit as st
from transformers import pipeline
# Load the model
st.title("PatientSeek AI Model")
st.write("This app uses the PatientSeek model for medical predictions.")
@st.cache_resource # Cache to speed up loading
def load_model():
return pipeline("text-classification", model="whyhow-ai/PatientSeek")
model = load_model()
# User input
user_input = st.text_area("Enter patient symptoms:")
if user_input:
result = model(user_input)
st.write("Prediction:", result)
|