use 5m_m2 variant

app.py

@@ -8,7 +8,7 @@ os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_cl
 
 
 import torch
-model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m1", pretrained=True)
+model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
 # or any of these variants
 # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
 # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_9m_m1", pretrained=True)
@@ -56,7 +56,7 @@ inputs = gr.inputs.Image(type='pil')
 outputs = gr.outputs.Label(type="confidences",num_top_classes=5)
 
 title = "SimpleNet"
-description = "Gradio demo for SimpleNet network pre-trained on ImageNet. This uses the simplenet_5m_m1 variant. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
+description = "Gradio demo for SimpleNet network pre-trained on ImageNet. This uses the simplenet_5m_m2 variant. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
 
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1608.06037' target='_blank'>Lets Keep it simple, Using simple architectures to outperform deeper and more complex architectures</a> | <a href='https://github.com/Coderx7/SimpleNet_Pytorch/blob/master/imagenet/simplenet.py' target='_blank'>Github Repo</a></p>"