Spaces:
Running
on
L40S
Running
on
L40S
Upload app_hg.py with huggingface_hub
Browse files
app_hg.py
CHANGED
|
@@ -21,6 +21,7 @@
|
|
| 21 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
| 22 |
import spaces
|
| 23 |
import os
|
|
|
|
| 24 |
import warnings
|
| 25 |
import argparse
|
| 26 |
import gradio as gr
|
|
|
|
| 21 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
| 22 |
import spaces
|
| 23 |
import os
|
| 24 |
+
os.environ['CUDA_HOME'] = '/usr/local/cuda'
|
| 25 |
import warnings
|
| 26 |
import argparse
|
| 27 |
import gradio as gr
|