Gradio

README.md

@@ -1,5 +1,5 @@
 ---
-title: Sample
+title: image-classification-with-explainability
 emoji: 📉
 colorFrom: blue
 colorTo: gray
@@ -11,3 +11,4 @@ license: bsd
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<!-- janakg/image-classification-with-explainability -->

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import numpy as np
 import torch
-from torchvision import models, transforms
 from PIL import Image
+from models.custom_resnet import CustomResNet
 
 
 # Run Interface
@@ -23,6 +23,11 @@ def run_inference(input_image, gradcam=False, gradcam_layer=3, gradcam_num = 3,
   """
 #   # Load the CIFAR-10 model
 
+  # model = CustomResNet()
+  # checkpoint = torch.load('weight/epoch=2-step=294.ckpt')
+  # model.load_state_dict(checkpoint['model_state_dict'])
+
+
 #   # Classify the image
 #   prediction = model.predict(image)
 #   predicted_class = np.argmax(prediction)

weight/epoch=2-step=294.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0470be596783dca3901dff62bc15b636b32ee6e707b5ddfa95b8562d7ae5c78b
+size 78937215