Update app.py
Browse files
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
|