Spaces:

plaidam
/

The_Broccolator

Running on Zero

plaidam commited on 2 days ago

Commit

c0abc9d

verified ·

1 Parent(s): d8c6c18

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,6 +9,8 @@ import torch
 import gradio as gr
 import spaces  # for ZeroGPU usage
 from typing import Sequence, Mapping, Any, Union
 import subprocess

 import gradio as gr
 import spaces  # for ZeroGPU usage
 from typing import Sequence, Mapping, Any, Union
+import os
+os.environ["CUDA_HOME"] = "/usr/local/cuda"
 import subprocess