Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,6 @@ def image_generation(model, number_of_images=1):
|
|
| 37 |
tile_pad=10,
|
| 38 |
pre_pad=0,
|
| 39 |
half=False,
|
| 40 |
-
device='cpu'
|
| 41 |
)
|
| 42 |
#TODO: img has to be same as opencv imread format
|
| 43 |
open_cv_image = np.array(img)
|
|
|
|
| 37 |
tile_pad=10,
|
| 38 |
pre_pad=0,
|
| 39 |
half=False,
|
|
|
|
| 40 |
)
|
| 41 |
#TODO: img has to be same as opencv imread format
|
| 42 |
open_cv_image = np.array(img)
|