Move natten install to app

app.py

@@ -1,6 +1,16 @@
+import subprocess
+import sys
+
 import gradio as gr
 from transformers import pipeline
 
+
+def install(package, index):
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package, index])
+
+
+install("natten", "-f https://shi-labs.com/natten/wheels/cpu/torch1.13/index.html")
+
 model_names = [
     "facebook/deit-base-patch16-224",
     "facebook/convnext-base-224",
@@ -10,7 +20,7 @@ model_names = [
     "microsoft/beit-base-patch16-224",
     "nvidia/mit-b0",
     "shi-labs/nat-base-in1k-224",
-    "shi-labs/dinat-base-in1k-224"
+    "shi-labs/dinat-base-in1k-224",
 ]
 
 
@@ -19,10 +29,13 @@ def process(image_file, top_k, model_name):
     pred = p(image_file)
     return {x["label"]: x["score"] for x in pred[:top_k]}
 
+
 # Inputs
 image = gr.Image(type="filepath", label="Upload an image")
 top_k = gr.Slider(minimum=1, maximum=10, step=1, value=5, label="Top k classes")
-model_selection = gr.Dropdown(model_names, label="Pick a model")
+model_selection = gr.Dropdown(
+    model_names, value="google/vit-base-patch16-224", label="Pick a model"
+)
 
 # Output
 labels = gr.Label()
@@ -36,9 +49,9 @@ iface = gr.Interface(
     inputs=[image, top_k, model_selection],
     outputs=[labels],
     examples=[
-	["bike.jpg", 5, "google/vit-base-patch16-224"],
-	["car.jpg", 5, "microsoft/swin-base-patch4-window7-224"],
-	["food.jpg", 5, "facebook/convnext-base-224"]
+        ["bike.jpg", 5, "google/vit-base-patch16-224"],
+        ["car.jpg", 5, "microsoft/swin-base-patch4-window7-224"],
+        ["food.jpg", 5, "facebook/convnext-base-224"],
     ],
     allow_flagging="never",
 )

requirements.txt

@@ -1,3 +1,2 @@
 torch==1.13.1
 transformers>=4.25.1
-natten -f https://shi-labs.com/natten/wheels/cpu/torch1.13/index.html