Sayiqa7 commited on
Commit
fbee908
·
verified ·
1 Parent(s): 073e2b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -2
app.py CHANGED
@@ -1,7 +1,33 @@
1
  import subprocess
2
- subprocess.check_call(["pip", "install", "transformers==4.34.0"])
3
  subprocess.check_call(["pip", "install", "torch>=1.7.1"])
4
- #Tokenizers 0.15.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
6
  import gradio as gr
7
 
 
1
  import subprocess
2
+ subprocess.check_call(["pip", "install", "transformers==4.35.2"])
3
  subprocess.check_call(["pip", "install", "torch>=1.7.1"])
4
+ subprocess.check_call(["pip", "install", "huggingface_hub>=0.19.0"])
5
+ subprocess.check_call(["pip", "install", "tokenizers>=0.15.0"])
6
+ subprocess.check_call(["pip", "install", "pytube"])
7
+ subprocess.check_call(["pip", "install", "pathlib"])
8
+ import transformers
9
+ import torch
10
+ import os
11
+ import pathlib
12
+ import pytube
13
+ def install_missing_packages():
14
+ required_packages = {
15
+ "torch":">=1.11.0",
16
+ "transformers":">=4.35.2",
17
+ "pytube":None,
18
+ "huggingface_hub": ">=0.19.0"
19
+
20
+ }
21
+
22
+
23
+ for package, version in required_packages.items():
24
+ try:
25
+ __import__(package)
26
+ except ImportError:
27
+ package_name = f"{package}{version}" if version else package
28
+ subprocess.check_call(["pip", "install", package_name])
29
+
30
+ install_missing_packages()
31
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
32
  import gradio as gr
33