Update app.py

app.py

@@ -11,8 +11,8 @@ edges2ferrets = './edges2ferrets_model.h5'
 ferretMaker = tf.keras.models.load_model(edges2ferrets)
 
 def transform_image(img, input_type):
-    sigma = 1.0  # Placeholder value for sigma
-    if input_type == "photo":
+    sigma = 1.7  # Placeholder value for sigma
+    if input_type == "Photo":
         # Resize the input photo to 256x256
         img = cv2.resize(img, (256, 256))
 
@@ -23,7 +23,7 @@ def transform_image(img, input_type):
         edges = (feature.canny(grey_img, sigma=sigma) * 255).astype(np.uint8)
         processed_input = cv2.cvtColor(edges, cv2.COLOR_GRAY2RGB)
 
-    elif input_type == "drawing":
+    elif input_type == "Drawing":
         # Resize the input drawing to 256x256
         img = cv2.resize(img, (256, 256))