Update app.py

app.py

@@ -8,6 +8,7 @@ from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image
 from resnet_lightning import ResNet18Model
 import gradio as gr
+from gradio import Inference
 
 model = ResNet18Model.load_from_checkpoint("epoch=19-step=3920.ckpt")
 
@@ -55,7 +56,7 @@ def inference(input_img, transparancy = 0.5, target_layer_number = -1):
     
     return classes[prediction[0].item(),visualization,confidences]
     
-demo = gr.Inference(
+demo = Inference(
     inference,
     inputs = [
         gr.Image(width=256,height=256,label="input image"),