fixing error message with cellpose model download

Author: carsen-stringer

suite2p/detection/detect.py

@@ -154,11 +154,12 @@ def detection_wrapper(f_reg, mov=None, yrange=None, xrange=None,
 
 	ymin = int(yrange[0])
 	xmin = int(xrange[0])
-	for s in stat:
-		s['ypix'] += ymin
-		s['xpix'] += xmin
-		s['med'][0] += ymin
-		s['med'][1] += xmin    
+	if len(stat) > 0:
+		for s in stat:
+			s['ypix'] += ymin
+			s['xpix'] += xmin
+			s['med'][0] += ymin
+			s['med'][1] += xmin    
 
 
 	if ops['preclassify'] > 0:

suite2p/gui/rungui.py

@@ -2,9 +2,12 @@
 from datetime import datetime
 
 import numpy as np
+
 from PyQt5 import QtGui, QtCore
 from PyQt5.QtWidgets import QDialog, QLineEdit, QLabel, QPushButton, QWidget, QGridLayout, QButtonGroup, QComboBox, QTextEdit, QFileDialog
 
+from cellpose.models import get_user_models, model_path, MODEL_NAMES
+
 from . import io
 from .. import default_ops
 
@@ -389,6 +392,10 @@ def compile_ops_db(self):
     def run_S2P(self):
         if len(self.opslist)==0:
             self.add_ops()
+        # pre-download model
+        pretrained_model_string = self.ops.get('pretrained_model', 'cyto')
+        pretrained_model_string = pretrained_model_string if pretrained_model_string is not None else 'cyto'
+        pretrained_model_path = model_path(pretrained_model_string, 0, True)
         self.finish = True
         self.error = False
         ops_file = os.path.join(self.ops_path, 'ops.npy')