jozee commited on
Commit
deeec20
·
verified ·
1 Parent(s): 5408e4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from PIL import Image
3
  import torch
4
  from transformers import SamModel, SamProcessor
5
  import numpy as np
 
 
6
  import matplotlib.pyplot as plt
7
 
8
  #Load the SAM model and processor
 
3
  import torch
4
  from transformers import SamModel, SamProcessor
5
  import numpy as np
6
+ import matplotlib
7
+ matplotlib.use("Agg")
8
  import matplotlib.pyplot as plt
9
 
10
  #Load the SAM model and processor