Kutluhan commited on
Commit
f6f32f9
·
1 Parent(s): 2d7c291

Spaces support is added.

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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="",