Update app.py
Browse files
app.py
CHANGED
|
@@ -24,9 +24,12 @@ from subprocess import call
|
|
| 24 |
os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "tmp")
|
| 25 |
cwd = os.getcwd()
|
| 26 |
print("Current working directory:", cwd)
|
| 27 |
-
|
| 28 |
# Installing dependencies not in requirements.txt
|
|
|
|
| 29 |
def install_add_dependencies():
|
|
|
|
|
|
|
| 30 |
with open('./build_ops.sh', 'rb') as file:
|
| 31 |
script = file.read()
|
| 32 |
return call(script, shell=True)
|
|
|
|
| 24 |
os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "tmp")
|
| 25 |
cwd = os.getcwd()
|
| 26 |
print("Current working directory:", cwd)
|
| 27 |
+
|
| 28 |
# Installing dependencies not in requirements.txt
|
| 29 |
+
@spaces.GPU
|
| 30 |
def install_add_dependencies():
|
| 31 |
+
print("inside install_add_dependencies")
|
| 32 |
+
print(torch.cuda.is_available())
|
| 33 |
with open('./build_ops.sh', 'rb') as file:
|
| 34 |
script = file.read()
|
| 35 |
return call(script, shell=True)
|