[martin-dev] fix space gpu
Browse files- demo/launch_gradio.py +3 -0
demo/launch_gradio.py
CHANGED
|
@@ -3,6 +3,8 @@
|
|
| 3 |
from typing import Any, Dict, List, Optional, Tuple
|
| 4 |
|
| 5 |
import gradio as gr
|
|
|
|
|
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
import numpy as np
|
| 8 |
import torch
|
|
@@ -534,6 +536,7 @@ def process_dual_inputs(
|
|
| 534 |
return None, f'Error: {str(e)}'
|
| 535 |
|
| 536 |
|
|
|
|
| 537 |
def create_demo() -> gr.Blocks:
|
| 538 |
"""Create and configure the Gradio demo interface for dual image comparison.
|
| 539 |
|
|
|
|
| 3 |
from typing import Any, Dict, List, Optional, Tuple
|
| 4 |
|
| 5 |
import gradio as gr
|
| 6 |
+
from spaces import GPU
|
| 7 |
+
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
import numpy as np
|
| 10 |
import torch
|
|
|
|
| 536 |
return None, f'Error: {str(e)}'
|
| 537 |
|
| 538 |
|
| 539 |
+
@GPU
|
| 540 |
def create_demo() -> gr.Blocks:
|
| 541 |
"""Create and configure the Gradio demo interface for dual image comparison.
|
| 542 |
|