File size: 1,039 Bytes
cb1857e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
### Imports for Modules ###
import gradio as gr
import os
import torch
from typing import Tuple, Dict
from timeit import default_timer as timer
### Functional Imports
from predictor import predictionMaker
exampleList = [["examples/" + example] for example in os.listdir("examples")]
title = "Detecting Retinal Diseases for Early Prevention"
description = "An EfficientNetB2 feature extractor computer vision model to classify OCT images into Brain Tumor types: CNV, DME, Drusen and Normal"
article = "Created by [Eternal Bliassard](https://github.com/EternalBlissard)."
# Create the Gradio demo
demo = gr.Interface(fn=predictionMaker,
inputs=[gr.Image(type="pil")],
outputs=[gr.Label(num_top_classes=2, label="Predictions"),
gr.Number(label="Prediction time (s)")],
examples=exampleList,
title=title,
description=description,
article=article)
# Launch the demo!
demo.launch()
|