Update app.py

app.py

@@ -9,6 +9,12 @@ import cv2
 from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 from pytorch_grad_cam.utils.image import show_cam_on_image
+import torch
+import cv2
+import numpy as np
+from torchvision import transforms
+from PIL import Image
+import gradio as gr
 import warnings
 warnings.filterwarnings("ignore")
 
@@ -76,10 +82,10 @@ def predict(input_image:Image.Image):
 interface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.inputs.Image(label="Input Image", type="pil")
+        gr.Image(label="Input Image", type="pil")
     ],
     outputs=[
-        gr.outputs.Label(label="Class"),
-        gr.outputs.Image(label="Face with Explainability", type="pil")
+        gr.Label(label="Class"),
+        gr.Image(label="Face with Explainability", type="pil")
     ],
 ).launch()