gouri100 commited on
Commit
e5412bd
·
verified ·
1 Parent(s): 36255e5

installing dependenies

Browse files
Files changed (1) hide show
  1. app.py +24 -5
app.py CHANGED
@@ -4,13 +4,32 @@ import numpy as np
4
  import io
5
  import pandas as pd
6
 
7
- import os
8
  import subprocess
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
- try:
11
- import ultralytics
12
- except ImportError:
13
- subprocess.run(["pip", "install", "ultralytics"])
14
 
15
  # Import your models and utility functions
16
  from TD import TD_model1
 
4
  import io
5
  import pandas as pd
6
 
 
7
  import subprocess
8
+ import sys
9
+
10
+ # List of required packages
11
+ required_packages = [
12
+ "transformers",
13
+ "ultralytics",
14
+ "opencv-python-headless",
15
+ "tqdm",
16
+ "torch",
17
+ "torchvision",
18
+ "easyocr"
19
+ ]
20
+
21
+ # Function to install packages
22
+ def install_packages(packages):
23
+ for package in packages:
24
+ try:
25
+ __import__(package.split("==")[0]) # Check if the package is already installed
26
+ except ImportError:
27
+ print(f"Installing {package}...")
28
+ subprocess.check_call([sys.executable, "-m", "pip", "install", package])
29
+
30
+ # Install required packages
31
+ install_packages(required_packages)
32
 
 
 
 
 
33
 
34
  # Import your models and utility functions
35
  from TD import TD_model1