Spaces:
Runtime error
Runtime error
rohithk-03
commited on
Commit
·
d020327
1
Parent(s):
9a973b8
update gdrive url
Browse files
model.py
CHANGED
@@ -128,6 +128,7 @@ def check_file(image_path):
|
|
128 |
output = ""
|
129 |
|
130 |
def test_model(model, test_loader, device):
|
|
|
131 |
model.to(device)
|
132 |
model.eval()
|
133 |
with torch.no_grad():
|
|
|
128 |
output = ""
|
129 |
|
130 |
def test_model(model, test_loader, device):
|
131 |
+
global output
|
132 |
model.to(device)
|
133 |
model.eval()
|
134 |
with torch.no_grad():
|