Spaces:
Running
on
Zero
Running
on
Zero
add space code
Browse files
app.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
from typing import Optional
|
|
|
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
import numpy as np
|
|
@@ -55,6 +56,7 @@ DEVICE = torch.device('cuda')
|
|
| 55 |
# @spaces.GPU
|
| 56 |
# @torch.inference_mode()
|
| 57 |
# @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
|
|
|
| 58 |
def process(
|
| 59 |
image_input,
|
| 60 |
box_threshold,
|
|
|
|
| 1 |
from typing import Optional
|
| 2 |
+
import spaces
|
| 3 |
|
| 4 |
import gradio as gr
|
| 5 |
import numpy as np
|
|
|
|
| 56 |
# @spaces.GPU
|
| 57 |
# @torch.inference_mode()
|
| 58 |
# @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
| 59 |
+
@spaces.GPU(duration=65)
|
| 60 |
def process(
|
| 61 |
image_input,
|
| 62 |
box_threshold,
|