app.py

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)