HyperX-Sen commited on
Commit
fb63550
·
verified ·
1 Parent(s): 43c2620

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -1,3 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
1
+ import subprocess
2
+ import sys
3
+
4
+ # Function to install dependencies if missing
5
+ def install(package):
6
+ subprocess.check_call([sys.executable, "-m", "pip", "install", package])
7
+
8
+ # List of required libraries
9
+ required_packages = ["transformers", "torch", "accelerate", "gradio", "huggingface_hub"]
10
+
11
+ # Install any missing packages
12
+ for package in required_packages:
13
+ try:
14
+ __import__(package)
15
+ except ImportError:
16
+ install(package)
17
+
18
  import gradio as gr
19
  import torch
20
  from transformers import AutoModelForCausalLM, AutoTokenizer