Update app.py

app.py

@@ -33,8 +33,10 @@ def segment(img, clases):
     prompts = clases.split(',')
 
     inputs = processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
+
     
-    outputs = model(**inputs)
+    with torch.no_grad():
+        outputs = model(**inputs)
     preds = outputs.logits.unsqueeze(1)
 
     return visualize_segmentation(image, prompts, preds)