Update app.py

app.py

@@ -6,18 +6,18 @@ def filter_img(input_img,filter_type):
     img = input_img
     filtered_img = np.zeros(img.shape, dtype=int)
     
-    if filter_type == 'vintage':
+    if filter_type == 'Vintage':
         sepia_filter = np.array(
             [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
         )
         filtered_img = img.dot(sepia_filter.T)
         filtered_img /= filtered_img.max()
-    elif filter_type == 'gray-sharp':
+    elif filter_type == 'Gray-sharp':
         img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
         filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
-    elif filter_type == 'sharp':
+    elif filter_type == 'Sharp':
         filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
-    elif filter_type == 'gray':
+    elif filter_type == 'Gray':
         filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
     elif filter_type == 'B&W':
         filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
@@ -41,7 +41,7 @@ def filter_img(input_img,filter_type):
     return filtered_img
 
 demo = gr.Interface(filter_img,
-                        inputs = [ gr.Image(),gr.inputs.Dropdown(["sharp","gray-sharp","vintage"])], 
+                        inputs = [ gr.Image(),gr.inputs.Dropdown(["Gray","B&W","Sharp","Gray-sharp","Vintage","Magenta","Yellow","Cyan","Blue","Green","Red"])], 
                         outputs = "image", 
                         description="Cool image filters!"
                     )