Update app.py
Browse files
app.py
CHANGED
|
@@ -5,4 +5,6 @@ pipe = pipeline("image-classification", model="skyau/dog-breed-classifier-vit")
|
|
| 5 |
img = st.file_uploader("image")
|
| 6 |
|
| 7 |
if img:
|
| 8 |
-
|
|
|
|
|
|
|
|
|
| 5 |
img = st.file_uploader("image")
|
| 6 |
|
| 7 |
if img:
|
| 8 |
+
img = Image.open(img)
|
| 9 |
+
label = pipe(img)
|
| 10 |
+
st.success(f"Is it: {label}?")
|