rohithk-03 commited on
Commit
d020327
·
1 Parent(s): 9a973b8

update gdrive url

Browse files
Files changed (1) hide show
  1. model.py +1 -0
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():