Update app.py

app.py

@@ -31,8 +31,8 @@ def download_weights():
               fid.write(r.content)
 
 try:
-    #download_weights()
-    model = models.CellposeModel(gpu=True, pretrained_model="cyto3")
+    download_weights()
+    model = models.CellposeModel(gpu=True, pretrained_model="cpsam")
 except Exception as e:
     print(f"Error loading model: {e}")
     exit(1)
@@ -128,22 +128,22 @@ def image_resize(img, resize=400):
     
 @spaces.GPU(duration=10)
 def run_model_gpu(img):
-    masks, flows, _ = model.eval(img, channels = [0,0])
+    masks, flows, _ = model.eval(img)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=60)
 def run_model_gpu60(img):
-    masks, flows, _ = model.eval(img, channels = [0,0])
+    masks, flows, _ = model.eval(img)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=240)
 def run_model_gpu240(img):
-    masks, flows, _ = model.eval(img, channels = [0,0])
+    masks, flows, _ = model.eval(img)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=1000)
 def run_model_gpu1000(img):
-    masks, flows, _ = model.eval(img, channels = [0,0])
+    masks, flows, _ = model.eval(img)#, channels = [0,0])
     return masks, flows
 
 #@spaces.GPU(duration=10)
@@ -227,7 +227,7 @@ with gr.Blocks(title = "Hello",
     with gr.Row():
         with gr.Column(scale=2):
             gr.HTML("""<div style="font-family:'Times New Roman', 'Serif'; font-size:16pt; font-weight:bold; text-align:center; color:white;">Cellpose-SAM for cellular segmentation</div>""")
-            gr.HTML("""<h4 style="color:white;">You may need to refresh/login for 5 minutes of free GPU compute time/day. </h4>""")
+            gr.HTML("""<h4 style="color:white;">You may need to refresh/login for 5 minutes of free GPU compute/day. </h4>""")
             gr.HTML("""<h4 style="color:white;">"pip install cellpose" for full functionality. </h4>""")
 
             input_image = gr.Image(label = "Input image", type = "filepath")