added basic gradio interface

app.py

@@ -1,10 +1,4 @@
-import gradio as gr
-
-import pandas as pd
-import numpy as np
-
 import torch
-# from torch.utils.data import Dataset, DataLoader, Subset
 from torch import nn
 from torchvision import models
 import torch.nn.functional as F
@@ -12,9 +6,6 @@ from torchvision.transforms import v2
 
 from PIL import Image
 
-# from PIL import ImageFile
-# ImageFile.LOAD_TRUNCATED_IMAGES = True
-
 labels = ['Pastel',
             'Yellow Belly',
             'Enchi',
@@ -79,5 +70,11 @@ def predict(img):
 
     return prediction_dict
 
+import gradio as gr
 
+gr.Interface(fn=predict,
+             inputs=gr.Image(type="pil"),
+             outputs=gr.Label(),
+             examples=["pastel_yb.jpg", "piebald.jpg", "leopard_fire.png"]
+             ).launch()