save images as PNG

app.py

@@ -6,15 +6,16 @@ import urllib.request
 import numpy as np
 import matplotlib.pyplot as plt
 from PIL import Image
+import subprocess
 
 def calculate_depth(model_type, img):
 
     if not os.path.exists('temp'):
       os.system('mkdir temp')
 
-    filename = "Images/Input-Test/0.jpg"
+    filename = "Images/Input-Test/0.png"
 
-    img.save(filename, "JPEG")
+    img.save(filename, "PNG")
 
     midas = torch.hub.load("intel-isl/MiDaS", model_type)
     
@@ -48,16 +49,16 @@ def calculate_depth(model_type, img):
 
     formatted = (output * 255.0 / np.max(output)).astype('uint8')
     out_im = Image.fromarray(formatted)
-    out_im.save("Images/Input-Test/0_d.jpg", "JPEG")
+    out_im.save("Images/Input-Test/0_d.png", "PNG")
 
     return out_im
 
 def wiggle_effect(slider):
 
-    return [f'Images/Input-Test/0_d.jpg',f'Images/Input-Test/0_d.jpg']
+    #subprocess.run(['echo', 'arg1'])
 
+    return [f'Images/Input-Test/1_d.png',f'Images/Input-Test/1_d.png']
 
-#pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
 
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")