fix jpeg error

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from PIL import Image
 
-def update(slider, img):
+def update(slider, model_type, img):
 
     if not os.path.exists('temp'):
       os.system('mkdir temp')
@@ -16,7 +16,7 @@ def update(slider, img):
 
     img.save(filename, "JPEG")
 
-    model_type = "DPT_Hybrid" 
+    #model_type = "DPT_Hybrid" 
     midas = torch.hub.load("intel-isl/MiDaS", model_type)
     
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
@@ -49,13 +49,8 @@ def update(slider, img):
 
     formatted = (output * 255 / np.max(output)).astype('uint8')
     out_im = Image.fromarray(formatted)
-
-    #out_im = Image.fromarray(output)
     out_im.save("temp/image_depth.jpeg", "JPEG")
 
-    #cv2.imwrite("temp/image_depth.jpeg", output)
-    #plt.imsave("temp/image_depth.jpeg", output)
-
     return f'temp/image_depth.jpeg'
 
 
@@ -64,6 +59,10 @@ with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")
     inp = [gr.Slider(1,15, default = 2, label='StepCycles',step= 1)]
 
+    midas_models = ["DPT_Large","DPT_Hybrid","MiDaS_small"]
+
+    inp.append(gr.inputs.Dropdown(midas_models, default="MiDaS_small", label="Midas_model_type"))
+
     with gr.Row():
         inp.append(gr.Image(type="pil", label="Input"))
         out = gr.Image(type="file", label="Output")