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 |
):
|