IMvision12 commited on
Commit
97bd2a2
·
1 Parent(s): 41bfe9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ def Predict(model, num_images):
32
  n += 1
33
  return images
34
 
35
- def inference(num_images, select: str):
36
- if select == 'Fashion_mnist':
37
  result = Predict(model2, num_images)
38
  else:
39
  result = Predict(model1, num_images)
 
32
  n += 1
33
  return images
34
 
35
+ def inference(num_images, Choose: str):
36
+ if Choose == 'Fashion_mnist':
37
  result = Predict(model2, num_images)
38
  else:
39
  result = Predict(model1, num_images)