Niraj70194 commited on
Commit
8de0612
·
verified ·
1 Parent(s): c798906

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def apply_gaussian_blur(image):
20
 
21
  # Perform semantic segmentation using the model
22
  with torch.no_grad():
23
- outputs = model(**inputs)
24
  logits = outputs.logits
25
 
26
  # Get the predicted class for each pixel
 
20
 
21
  # Perform semantic segmentation using the model
22
  with torch.no_grad():
23
+ outputs = segmentation_model(**inputs)
24
  logits = outputs.logits
25
 
26
  # Get the predicted class for each pixel