Update app.py

app.py

@@ -11,13 +11,13 @@ input_size = (256, 256, 1)
 def process_image(task, deg_image):
     if task == 'binarize':
         generator = generator_model(biggest_layer=1024)
-        generator.load_weights("weights/binarization_generator_weights.h5")
+        generator.load_weights("binarization_generator_weights.h5")
     elif task == 'deblur':
         generator = generator_model(biggest_layer=1024)
-        generator.load_weights("weights/deblur_weights.h5")
+        generator.load_weights("deblur_weights.h5")
     elif task == 'unwatermark':
         generator = generator_model(biggest_layer=512)
-        generator.load_weights("weights/watermark_rem_weights.h5")
+        generator.load_weights("watermark_rem_weights.h5")
     else:
         return "Wrong task, please specify a correct task!"