arpandesign commited on
Commit
03c24b9
·
verified ·
1 Parent(s): 4613e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,13 +72,13 @@ model_0 = NumberClassifier(input_shape=1, # number of color channels (3 for RGB)
72
  hidden_units=10,
73
  output_shape=10).to(device)
74
 
75
- model_0.load_state_dict(torch.load("/home/arpan/torchenv_learning/models/pytorch_num_classifier_final_model_with_EMNIST.pth"))
76
 
77
 
78
  title = "Number Classifier Minimal"
79
  description = "An Image feature extractor computer vision model to classify images of handwritten digits."
80
  article = "Created at [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/)."
81
- example_list = [[str(filepath)] for filepath in random.sample(glob("/home/arpan/torchenv_learning/showcase_dataset/*"), k=25)]
82
  example_list
83
 
84
 
 
72
  hidden_units=10,
73
  output_shape=10).to(device)
74
 
75
+ model_0.load_state_dict(torch.load("pytorch_num_classifier_final_model_with_EMNIST.pth"))
76
 
77
 
78
  title = "Number Classifier Minimal"
79
  description = "An Image feature extractor computer vision model to classify images of handwritten digits."
80
  article = "Created at [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/)."
81
+ example_list = [[str(filepath)] for filepath in random.sample(glob("examples/*"), k=25)]
82
  example_list
83
 
84