Update app.py

app.py

@@ -5,4 +5,28 @@ description="Gradio Demo for Swin Transformer: Hierarchical Vision Transformer u
 
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2103.14030' target='_blank'>Swin Transformer: Hierarchical Vision Transformer using Shifted Windows</a> | <a href='https://github.com/microsoft/Swin-Transformer' target='_blank'>Github Repo</a></p>"
 
-gr.Interface.load("huggingface/microsoft/swin-large-patch4-window12-384-in22k",title=title, article=article,description=description,examples=[["tiger.jpeg"]]).launch(enable_queue=True,cache_examples=True)
+gr.Interface.load("huggingface/microsoft/swin-large-patch4-window12-384-in22k",title=title, article=article,description=description,examples=[["tiger.jpeg"]]).launch(enable_queue=True,cache_examples=True)
+
+io1 = gr.Interface.load("huggingface/microsoft/swin-large-patch4-window12-384-in22k")
+
+io2 = gr.Interface.load("huggingface/microsoft/swin-base-patch4-window7-224-in22k")
+
+
+def inference(image, model):
+    if model == "swin-large-patch4-window12-384-in22k":
+        outtext = io1(image)
+    else:
+        outtext = io2(image)
+    return outtext   
+    
+     
+examples=[['tiger.jpeg',"swin-large-patch4-window12-384-in22k"]]
+gr.Interface(
+    inference, 
+    [gr.inputs.Image(label="Input Image"),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, cache_examples=True)