Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def load_img (filename):
|
|
21 |
return img
|
22 |
|
23 |
def process_img(image):
|
24 |
-
y = image
|
25 |
|
26 |
with torch.no_grad():
|
27 |
result = model(y)
|
|
|
21 |
return img
|
22 |
|
23 |
def process_img(image):
|
24 |
+
y = image#.to(device)
|
25 |
|
26 |
with torch.no_grad():
|
27 |
result = model(y)
|