Update app.py

app.py

@@ -2,15 +2,12 @@ from openvino.runtime import Core
 import gradio as gr 
 import numpy as np 
 from PIL import Image
+from torchvision import models,transforms
 
-def normalize(img):
-  img=img.astype(np.float32)
-  mean=(0.485, 0.456, 0.406)
-  std=(0.229, 0.224, 0.225)
-  img/=255.0
-  img-=mean
-  img/std
-  return img
+tfms = transforms.Compose([
+    transforms.ToTensor(),
+    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) # imagenet
+])
 def segment_image(filepath):
   image=cv2.imread(filepath)
   image=cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
@@ -26,8 +23,8 @@ def segment_image(filepath):
 
 demo=gr.Interface(fn=segment_image,inputs=gr.Image(type='filepath'),
                   outputs=[gr.Image(type="pil"),gr.Image(type="pil")],
-                  examples=['R001_ch1_video_03_00-29-13-03.jpg',
-                            "R002_ch1_video_01_01-07-25-19.jpg,
+                  examples=["R001_ch1_video_03_00-29-13-03.jpg",
+                            "R002_ch1_video_01_01-07-25-19.jpg",
                             "R003_ch1_video_05_00-22-42-23.jpg",
                             "R004_ch1_video_01_01-12-22-00.jpg",
                             "R005_ch1_video_03_00-19-10-11.jpg",