aidoctor / app.py
HURUAI's picture
Update app.py
942c3fd
raw
history blame contribute delete
211 Bytes
import streamlit as st
from transformers import pipeline
pipe = pipeline('text-generation', model="gpt2")
text = st.text_area("Ask the Doctor a Question?")
if text:
out = pipe(text)
st.json(out)