hvoss-techfak commited on
Commit
0d4640a
·
1 Parent(s): 9f2e1e5

zero gpu support

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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
  ):