Update app.py

app.py

@@ -8,7 +8,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2103.140
 
 io1 = gr.Interface.load("huggingface/microsoft/swin-large-patch4-window12-384-in22k")
 
-#io2 = gr.Interface.load("huggingface/microsoft/swin-base-patch4-window7-224-in22k")
+io2 = gr.Interface.load("huggingface/microsoft/swin-base-patch4-window7-224-in22k")
 
 
 def inference(image, model):
@@ -23,10 +23,10 @@ examples=[['tiger.jpeg','swin-large-patch4-window12-384-in22k']]
 
 gr.Interface(
     inference, 
-    [gr.inputs.Image(label="Input Image",type='filepath'),gr.inputs.Dropdown(choices=["swin-large-patch4-window12-384-in22k"], type="value", default="swin-large-patch4-window12-384-in22k", label="model")
+    [gr.inputs.Image(label="Input Image",type='filepath'),gr.inputs.Dropdown(choices=["swin-large-patch4-window12-384-in22k","swin-base-patch4-window7-224-in22k"], type="value", default="swin-large-patch4-window12-384-in22k", label="model")
 ], 
     gr.outputs.Label(label="Classification"),
     examples=examples,
     article=article,
     title=title,
-    description=description).launch(enable_queue=True)
+    description=description).launch(enable_queue=True,cache_examples=True)