Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -96,7 +96,9 @@ def clean_img(image, mask):
|
|
| 96 |
image = Image.fromarray(image).convert('RGB')
|
| 97 |
return image
|
| 98 |
|
| 99 |
-
@spaces.GPU
|
|
|
|
|
|
|
| 100 |
def floorplan_generation(outline, num_of_rooms):
|
| 101 |
new_width = 512
|
| 102 |
new_height = 512
|
|
|
|
| 96 |
image = Image.fromarray(image).convert('RGB')
|
| 97 |
return image
|
| 98 |
|
| 99 |
+
# @spaces.GPU
|
| 100 |
+
@spaces.GPU(duration=80)
|
| 101 |
+
|
| 102 |
def floorplan_generation(outline, num_of_rooms):
|
| 103 |
new_width = 512
|
| 104 |
new_height = 512
|