Update app.py

app.py

@@ -5,10 +5,11 @@ import csv
 import time
 
 
+
+# Image Classification Using Torch:
 import requests
 from PIL import Image
 from torchvision import transforms
-
 # Download human-readable labels for ImageNet.
 response = requests.get("https://git.io/JJkYN")
 labels = response.text.split("\n")
@@ -21,14 +22,29 @@ def predict(inp):
   return confidences
 
 import gradio as gr
-
-gr.Interface(fn=predict, 
+interface1=gr.Interface(fn=predict, 
              inputs=gr.Image(type="pil"),
              outputs=gr.Label(num_top_classes=3),
-             examples=["lion.jpg", "cheetah.jpg"]).launch()
+             examples=["lion.jpg", "cheetah.jpg"])
 
+# Image Classification Using VIT - Vision Image Transformers Architecture
 
-    
+import gradio as gr
+interface2=gr.Interface.load(
+             "huggingface/google/vit-base-patch16-224",
+             examples=["alligator.jpg", "laptop.jpg"])
+
+
+# ONNX Model Zoo:
+
+gr.Markdown("""
+
+https://github.com/onnx/models
+
+""")
+
+interface1.launch()
+interface2.launch()
 
 
 uploaded_images = {'characters': {}, 'terrain': {}}