test update for png support

app.py

@@ -56,21 +56,19 @@ class GOSNormalize(object):
 transform =  transforms.Compose([GOSNormalize([0.5,0.5,0.5],[1.0,1.0,1.0])])
 
 @functools.lru_cache()
-def get_url_im(t):
+def get_url_im(url):
     user_agent = {'User-agent': 'gradio-app'}
-    response = requests.get(t, headers=user_agent)
-    return (BytesIO(response.content))
-
+    response = requests.get(url, headers=user_agent)
+    return BytesIO(response.content)
 
 def load_image(im_path, hypar):
-    # im_path = BytesIO(requests.get(im_path).content)
-    
     im_path = get_url_im(im_path)
-    im = im_reader(im_path)
-    im, im_shp = im_preprocess(im, hypar["cache_size"])
-    im = torch.divide(im,255.0)
+    im = Image.open(im_path)
+    im = im.convert("RGB")  # Convert image to RGB
+    im, im_shp = im_preprocess(np.array(im), hypar["cache_size"])
+    im = torch.divide(im, 255.0)
     shape = torch.from_numpy(np.array(im_shp))
-    return transform(im).unsqueeze(0), shape.unsqueeze(0) # make a batch of image, shape
+    return transform(im).unsqueeze(0), shape.unsqueeze(0)
 
 
 def build_model(hypar,device):