Spaces:
Runtime error
Runtime error
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,
|