Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,9 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
|
3 |
+
if torch.cuda.is_available() or os.getenv("ZERO_GPU_SUPPORT", False):
|
4 |
+
gr.load("models/infly/OpenCoder-8B-Instruct").launch()
|
5 |
+
else:
|
6 |
+
raise RuntimeError("No compatible GPU environment found for this model.")
|
7 |
+
|
8 |
+
#@spaces.GPU
|
9 |
+
|