Update app.py

app.py

@@ -112,6 +112,13 @@ API_dict = dict(
     SwinForImageClassification_24 = "https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099224",
     BeitForImageClassification_25="https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099225")
 
+pipeline_dict = dict(
+    SwinForImageClassification_21="rwcuffney/autotrain-pick_a_card-3726099221",
+    ViTForImageClassification_22="rwcuffney/autotrain-pick_a_card-3726099222",
+    ResNetForImageClassification_23= "rwcuffney/autotrain-pick_a_card-3726099223",
+    SwinForImageClassification_24 = rwcuffney/autotrain-pick_a_card-3726099224",
+    BeitForImageClassification_25="rwcuffney/autotrain-pick_a_card-3726099225")
+
 
 # printing enum member as string
 #print(Api_URL.ViTForImageClassification_22.value)
@@ -161,7 +168,7 @@ with st.form("api_form"):
 
     submitted = st.form_submit_button("Submit")
     if submitted:
-        pipeline = pipeline(task="image-classification", model="rwcuffney/autotrain-pick_a_card-3726099224")
+        pipeline = pipeline(task="image-classification", model=pipeline_dict[api])
         def predict(image):
             predictions = pipeline(image)
             return {p["label"]: p["score"] for p in predictions}