Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ import PIL
|
|
| 24 |
import torch.nn.functional as F
|
| 25 |
from torchvision import transforms
|
| 26 |
|
| 27 |
-
import spaces
|
| 28 |
from huggingface_hub import snapshot_download
|
| 29 |
|
| 30 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
@@ -122,7 +122,7 @@ def init_pipe():
|
|
| 122 |
|
| 123 |
return pipe
|
| 124 |
|
| 125 |
-
|
| 126 |
def inference(source_images,
|
| 127 |
target_images,
|
| 128 |
text_prompt, negative_prompt,
|
|
|
|
| 24 |
import torch.nn.functional as F
|
| 25 |
from torchvision import transforms
|
| 26 |
|
| 27 |
+
#import spaces
|
| 28 |
from huggingface_hub import snapshot_download
|
| 29 |
|
| 30 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 122 |
|
| 123 |
return pipe
|
| 124 |
|
| 125 |
+
#@spaces.GPU()#duration=120
|
| 126 |
def inference(source_images,
|
| 127 |
target_images,
|
| 128 |
text_prompt, negative_prompt,
|