Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -112,6 +112,13 @@ API_dict = dict(
|
|
112 |
SwinForImageClassification_24 = "https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099224",
|
113 |
BeitForImageClassification_25="https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099225")
|
114 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
115 |
|
116 |
# printing enum member as string
|
117 |
#print(Api_URL.ViTForImageClassification_22.value)
|
@@ -161,7 +168,7 @@ with st.form("api_form"):
|
|
161 |
|
162 |
submitted = st.form_submit_button("Submit")
|
163 |
if submitted:
|
164 |
-
pipeline = pipeline(task="image-classification", model=
|
165 |
def predict(image):
|
166 |
predictions = pipeline(image)
|
167 |
return {p["label"]: p["score"] for p in predictions}
|
|
|
112 |
SwinForImageClassification_24 = "https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099224",
|
113 |
BeitForImageClassification_25="https://api-inference.huggingface.co/models/rwcuffney/autotrain-pick_a_card-3726099225")
|
114 |
|
115 |
+
pipeline_dict = dict(
|
116 |
+
SwinForImageClassification_21="rwcuffney/autotrain-pick_a_card-3726099221",
|
117 |
+
ViTForImageClassification_22="rwcuffney/autotrain-pick_a_card-3726099222",
|
118 |
+
ResNetForImageClassification_23= "rwcuffney/autotrain-pick_a_card-3726099223",
|
119 |
+
SwinForImageClassification_24 = rwcuffney/autotrain-pick_a_card-3726099224",
|
120 |
+
BeitForImageClassification_25="rwcuffney/autotrain-pick_a_card-3726099225")
|
121 |
+
|
122 |
|
123 |
# printing enum member as string
|
124 |
#print(Api_URL.ViTForImageClassification_22.value)
|
|
|
168 |
|
169 |
submitted = st.form_submit_button("Submit")
|
170 |
if submitted:
|
171 |
+
pipeline = pipeline(task="image-classification", model=pipeline_dict[api])
|
172 |
def predict(image):
|
173 |
predictions = pipeline(image)
|
174 |
return {p["label"]: p["score"] for p in predictions}
|