Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
d3ab058
1
Parent(s):
ae7902f
fix logging
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from sentry_sdk import capture_exception, push_scope, capture_message
|
|
8 |
from sentry_sdk.integrations.logging import LoggingIntegration
|
9 |
from sentry_sdk.integrations.starlette import StarletteIntegration
|
10 |
from sentry_sdk.integrations.fastapi import FastApiIntegration
|
11 |
-
import spaces
|
12 |
dsn = os.getenv("SENTRY_DSN")
|
13 |
if not dsn:
|
14 |
print("WARNING: SENTRY_DSN not set – Sentry disabled")
|
@@ -310,7 +310,7 @@ if os.path.exists(DEFAULT_MATERIALS_CSV):
|
|
310 |
else:
|
311 |
initial_df.to_csv(DEFAULT_MATERIALS_CSV, index=False)
|
312 |
|
313 |
-
|
314 |
def run_autoforge_process(cmd, log_path):
|
315 |
"""
|
316 |
Launch the external `autoforge` CLI.
|
@@ -567,7 +567,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
567 |
)
|
568 |
gr.Markdown(
|
569 |
'Please note that huggingface enforces a maximum execution time of one minute. Depending on your configuration (especially iteration count) it is possible to exceed this time limit. In that case you will see a "GPU Task aborted" error.'
|
570 |
-
'If you need more time, take a look at the [Autoforge Github Page](https://github.com/hvoss-techfak/AutoForge) to see how you can run the program locally, or pull the docker container for this project (upper right corner -> three dots -> "run locally")'
|
571 |
)
|
572 |
|
573 |
with gr.Row():
|
@@ -836,7 +836,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
836 |
worker.join() # make sure it’s done
|
837 |
return_code = worker.returncode
|
838 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
839 |
if return_code != 0:
|
|
|
840 |
err = RuntimeError(f"Autoforge exited with code {return_code} \n {log_output}")
|
841 |
capture_exception(err) # send to Sentry
|
842 |
raise err # also fail the Gradio call
|
@@ -847,6 +857,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
847 |
)
|
848 |
log_str = " ".join(log_output)
|
849 |
|
|
|
|
|
850 |
files_to_offer = [
|
851 |
p
|
852 |
for p in [
|
|
|
8 |
from sentry_sdk.integrations.logging import LoggingIntegration
|
9 |
from sentry_sdk.integrations.starlette import StarletteIntegration
|
10 |
from sentry_sdk.integrations.fastapi import FastApiIntegration
|
11 |
+
#import spaces
|
12 |
dsn = os.getenv("SENTRY_DSN")
|
13 |
if not dsn:
|
14 |
print("WARNING: SENTRY_DSN not set – Sentry disabled")
|
|
|
310 |
else:
|
311 |
initial_df.to_csv(DEFAULT_MATERIALS_CSV, index=False)
|
312 |
|
313 |
+
#@spaces.GPU(duration=90) # GPU reserved only for this call
|
314 |
def run_autoforge_process(cmd, log_path):
|
315 |
"""
|
316 |
Launch the external `autoforge` CLI.
|
|
|
567 |
)
|
568 |
gr.Markdown(
|
569 |
'Please note that huggingface enforces a maximum execution time of one minute. Depending on your configuration (especially iteration count) it is possible to exceed this time limit. In that case you will see a "GPU Task aborted" error.'
|
570 |
+
' If you need more time, take a look at the [Autoforge Github Page](https://github.com/hvoss-techfak/AutoForge) to see how you can run the program locally, or pull the docker container for this project (upper right corner -> three dots -> "run locally")'
|
571 |
)
|
572 |
|
573 |
with gr.Row():
|
|
|
836 |
worker.join() # make sure it’s done
|
837 |
return_code = worker.returncode
|
838 |
|
839 |
+
try:
|
840 |
+
sentry_sdk.add_attachment(
|
841 |
+
path=log_file,
|
842 |
+
filename="autoforge.log",
|
843 |
+
content_type="text/plain",
|
844 |
+
)
|
845 |
+
except Exception as e:
|
846 |
+
capture_exception(e)
|
847 |
+
|
848 |
if return_code != 0:
|
849 |
+
|
850 |
err = RuntimeError(f"Autoforge exited with code {return_code} \n {log_output}")
|
851 |
capture_exception(err) # send to Sentry
|
852 |
raise err # also fail the Gradio call
|
|
|
857 |
)
|
858 |
log_str = " ".join(log_output)
|
859 |
|
860 |
+
|
861 |
+
|
862 |
files_to_offer = [
|
863 |
p
|
864 |
for p in [
|