JSun22 commited on
Commit
b7860a6
1 Parent(s): 9ea315f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ with open(r"labels.txt", "r") as f:
18
  def model_parameter(model):
19
  for name,param in model.named_parameters():
20
  print(name,param)
 
21
 
22
  def predict(inp):
23
  inp = transforms.ToTensor()(inp).unsqueeze(0)
 
18
  def model_parameter(model):
19
  for name,param in model.named_parameters():
20
  print(name,param)
21
+ return
22
 
23
  def predict(inp):
24
  inp = transforms.ToTensor()(inp).unsqueeze(0)