.

app.py

@@ -291,7 +291,7 @@ def plot_predictions(input_dict, preds, masks, image_size=224):
     plt.close()
 
 
-def inference(img, num_tokens, perform_sampling, num_rgb, num_depth, num_semseg, seed):
+def inference(img, num_tokens, manual_mode, num_rgb, num_depth, num_semseg, seed):
     im = Image.open(img)
 
     # Center crop and resize RGB
@@ -325,7 +325,7 @@ def inference(img, num_tokens, perform_sampling, num_rgb, num_depth, num_semseg,
     input_dict = {k: v.to(device) for k,v in input_dict.items()}
 
 
-    if perform_sampling:
+    if not manual_mode:
         # Randomly sample masks
 
         torch.manual_seed(int(time.time())) # Random mode is random
@@ -398,4 +398,4 @@ gr.Interface(
     description=description,
     article=article,
     examples=examples
-).launch(enable_queue=True, cache_examples=True)
+).launch(enable_queue=True, cache_examples=False)