Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,9 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
|
|
24 |
|
25 |
def transfer(image, model):
|
26 |
device = 'cpu'
|
27 |
-
|
|
|
|
|
28 |
width = image.size[0]
|
29 |
height = image.size[1]
|
30 |
|
|
|
24 |
|
25 |
def transfer(image, model):
|
26 |
device = 'cpu'
|
27 |
+
if model is None:
|
28 |
+
model = '0'
|
29 |
+
|
30 |
width = image.size[0]
|
31 |
height = image.size[1]
|
32 |
|