Update app.py

app.py

@@ -56,9 +56,9 @@ def _dynGPU(
 def dynGPU(
     fn: Callable[P, R] | None = None,
     duration: Callable[P, int] = lambda: 60,
-    min=30,
+    min=10,
     max=300,
-    step=10,
+    step=5,
 ) -> Callable[P, R]:
     if fn is None:
         return partial(_dynGPU, duration=duration, min=min, max=max, step=step)
@@ -260,9 +260,7 @@ def process_images(images, threshold):
             
             batch = batch.to(device) 
             with torch.no_grad():
-                logits = model(batch)
-                probabilities = torch.nn.functional.sigmoid(logits)
-            
+                probabilities = model(batch)[0]
                 for i, prob in enumerate(probabilities):
                     indices = torch.where(prob > threshold)[0]
                     values = prob[indices]