Ticoliro commited on
Commit
16cebe7
·
verified ·
1 Parent(s): 6f40c84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -1,3 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import spaces
2
  import gradio as gr
3
  import torch
 
1
+ import sys
2
+ import subprocess
3
+ import pkg_resources
4
+
5
+ # Função para verificar e instalar pacotes faltantes
6
+ def install_missing_packages():
7
+ required = {'torch', 'torchaudio', 'transformers', 'gradio', 'numpy', 'accelerate', 'soundfile'}
8
+ installed = {pkg.key for pkg in pkg_resources.working_set}
9
+ missing = required - installed
10
+
11
+ if missing:
12
+ python = sys.executable
13
+ subprocess.check_call([python, '-m', 'pip', 'install', *missing], stdout=subprocess.DEVNULL)
14
+
15
+ # Instalar pacotes faltantes
16
+ install_missing_packages()
17
+
18
  import spaces
19
  import gradio as gr
20
  import torch