Update app.py

app.py

@@ -4,7 +4,7 @@ import torchvision.transforms as transforms
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
-resneXt = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_resneXt')
+resneXt = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_se_resnext101_32x4d')
 utils = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_convnets_processing_utils')
 
 resneXt.eval().to(device)
@@ -34,10 +34,10 @@ def inference(img):
   
   return results
   
-title="ResNeXt101"
-description="Gradio demo for ResNeXt101, ResNet with bottleneck 3x3 Convolutions substituted by 3x3 Grouped Convolutions, trained with mixed precision using Tensor Cores. To use it, simply upload your image or click on one of the examples below. Read more at the links below"
+title="SE-ResNeXt101"
+description="Gradio demo for SE-ResNeXt101, ResNeXt with Squeeze-and-Excitation module added, trained with mixed precision using Tensor Cores. To use it, simply upload your image or click on one of the examples below. Read more at the links below"
 
-article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1611.05431'>Aggregated Residual Transformations for Deep Neural Networks</a> | <a href='https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnext101-32x4d'>Github Repo</a></p>"
+article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1709.01507'>Squeeze-and-Excitation Networks</a> | <a href='https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/se-resnext101-32x4d'>Github Repo</a></p>"
 
 examples=[['food.jpeg']]
 gr.Interface(inference,gr.inputs.Image(type="pil"),"text",title=title,description=description,article=article,examples=examples).launch(enable_queue=True)