Spaces:
Build error
Build error
Spaces support is added.
Browse files
app.py
CHANGED
|
@@ -19,9 +19,12 @@ import warnings
|
|
| 19 |
from torchvision import transforms
|
| 20 |
import torch.nn.functional as F
|
| 21 |
import huggingface_hub
|
|
|
|
| 22 |
|
| 23 |
model_path = huggingface_hub.hf_hub_download("Kutluhan/SemanticInpaint", "model008000.pt")
|
| 24 |
|
|
|
|
|
|
|
| 25 |
def create_argparser():
|
| 26 |
defaults = dict(
|
| 27 |
data_dir="",
|
|
|
|
| 19 |
from torchvision import transforms
|
| 20 |
import torch.nn.functional as F
|
| 21 |
import huggingface_hub
|
| 22 |
+
import spaces
|
| 23 |
|
| 24 |
model_path = huggingface_hub.hf_hub_download("Kutluhan/SemanticInpaint", "model008000.pt")
|
| 25 |
|
| 26 |
+
|
| 27 |
+
@spaces.GPU(duration=240)
|
| 28 |
def create_argparser():
|
| 29 |
defaults = dict(
|
| 30 |
data_dir="",
|