minor add

app.py

@@ -68,7 +68,7 @@ def get_model(model_name, pretrained=True):
     cached_path = f'cache/{model_name}_prototype.pt'  # noqa
     cfg.model.prototype = cached_path
 
-    model = init_model(metainfo.config, None, device='cuda')
+    model = init_model(metainfo.config, None, device='cpu')
     with st.spinner(f'Loading model {model_name} on the server...This is '
                     'slow at the first time.'):
         model.init_weights()
@@ -133,9 +133,14 @@ def app():
     model_name = st.sidebar.selectbox("Model:", ['resnet50_8xb32_in1k'])
 
     st.markdown(
-        "<h1 style='text-align: center;'>Image To Image Retrieval</h1>",
+        "<h1>Image To Image Retrieval</h1>",
         unsafe_allow_html=True,
     )
+    st.write(
+        'This is a demo for image to image retrieval in around 3k images from '
+        'ImageNet tiny val set using mmclassification apis. You can try more '
+        'features on [mmclassification]'
+        '(https://github.com/open-mmlab/mmclassification).')
 
     file = st.file_uploader(
         'Please upload your own image or use the provided:')

requirements.txt

@@ -1,4 +1,5 @@
 torch
+torchvision
 mmengine>=0.4.0
 mmcls>=1.0.0rc5
 mmcv>=2.0.0rc3