Spaces:
Running
on
Zero
Running
on
Zero
wondervictor
commited on
Update model_new.py
Browse files- model_new.py +1 -1
model_new.py
CHANGED
@@ -77,7 +77,7 @@ class Model:
|
|
77 |
|
78 |
@torch.no_grad()
|
79 |
@spaces.GPU(enable_queue=True)
|
80 |
-
def
|
81 |
self,
|
82 |
image: np.ndarray,
|
83 |
prompt: str,
|
|
|
77 |
|
78 |
@torch.no_grad()
|
79 |
@spaces.GPU(enable_queue=True)
|
80 |
+
def process_edge(
|
81 |
self,
|
82 |
image: np.ndarray,
|
83 |
prompt: str,
|