Update app.py

app.py

@@ -25,7 +25,17 @@ def check_forgery_df(img):
     x = np.expand_dims( img.astype('float32')/255., axis=0 )
     
     pred1 = model_M1.predict(x, verbose=0)
-    pred2= model_M2.predict(x, verbose=0)    
+    pred2= model_M2.predict(x, verbose=0)   
+
+
+    # Ensure pred1 and pred2 are numpy arrays before proceeding
+    if isinstance(pred1, dict):
+        print("pred1 is dict!")
+        pred1 = pred1[next(iter(pred1))]
+    if isinstance(pred2, dict):
+        pred2 = pred2[next(iter(pred2))]
+
+    
     pred = np.max([pred1,pred2], axis=0)   
     
     pred = dfutils.create_mask(pred)