import gradio as gr
from gradio.mix import Parallel
modelPneumonia = gr.Interface.load("huggingface/nickmuchi/vit-finetuned-chest-xray-pneumonia",
                                   title="Pneumonia Prediction", 
                                   description="Predict Pneumonia from XRay Image",
                                   article = "<h4>This model is a fine-tuned version of google/vit-base-patch16-224-in21k on the chest-xray-pneumonia dataset </h4>"
                                  )
modelPneumonia.close()
modelPneumonia.launch()