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