Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0d4640a
1
Parent(s):
9f2e1e5
zero gpu support
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ from sentry_sdk import capture_exception, push_scope, capture_message
|
|
| 6 |
from sentry_sdk.integrations.logging import LoggingIntegration
|
| 7 |
from sentry_sdk.integrations.starlette import StarletteIntegration
|
| 8 |
from sentry_sdk.integrations.fastapi import FastApiIntegration
|
|
|
|
| 9 |
dsn = os.getenv("SENTRY_DSN")
|
| 10 |
if not dsn:
|
| 11 |
print("WARNING: SENTRY_DSN not set – Sentry disabled")
|
|
@@ -600,6 +601,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 600 |
)
|
| 601 |
|
| 602 |
# --- Backend Function for Running the Script ---
|
|
|
|
| 603 |
def execute_autoforge_script(
|
| 604 |
current_filaments_df_state_val, input_image, *accordion_param_values
|
| 605 |
):
|
|
|
|
| 6 |
from sentry_sdk.integrations.logging import LoggingIntegration
|
| 7 |
from sentry_sdk.integrations.starlette import StarletteIntegration
|
| 8 |
from sentry_sdk.integrations.fastapi import FastApiIntegration
|
| 9 |
+
import spaces
|
| 10 |
dsn = os.getenv("SENTRY_DSN")
|
| 11 |
if not dsn:
|
| 12 |
print("WARNING: SENTRY_DSN not set – Sentry disabled")
|
|
|
|
| 601 |
)
|
| 602 |
|
| 603 |
# --- Backend Function for Running the Script ---
|
| 604 |
+
@spaces.GPU
|
| 605 |
def execute_autoforge_script(
|
| 606 |
current_filaments_df_state_val, input_image, *accordion_param_values
|
| 607 |
):
|