aixen commited on
Commit
1feca79
·
1 Parent(s): 14d79ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -3
app.py CHANGED
@@ -1,5 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  from transformers import pipeline
2
- generator = pipeline("mask-generation", device = 0, points_per_batch = 256)
3
- image_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"
4
- outputs = generator(image_url, points_per_batch = 256)
5
 
 
 
 
 
 
1
+ import importlib
2
+
3
+ # Check if transformers is already installed
4
+ try:
5
+ importlib.import_module('transformers')
6
+ print("transformers library is already installed")
7
+ except ImportError:
8
+ # transformers is not installed, so install it
9
+ import subprocess
10
+ subprocess.check_call(["pip", "install", "transformers"])
11
+ print("transformers library has been installed")
12
+
13
+ # Import the necessary modules
14
  from transformers import pipeline
 
 
 
15
 
16
+ # Use the transformers library
17
+ generator = pipeline("mask-generation", device=0, points_per_batch=256)
18
+ image_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"
19
+ outputs = generator(image_url, points_per_batch=256)