Update app.py
Browse files
app.py
CHANGED
@@ -37,13 +37,13 @@ st.sidebar.markdown("*Swin Transformer* https://huggingface.co/docs/transformers
|
|
37 |
# Image classification function
|
38 |
|
39 |
def classify_image1(image):
|
40 |
-
pipe1 = pipeline("image-classification", "SolubleFish/swin_transformer-finetuned-eurosat"
|
41 |
return pipe1(image)
|
42 |
def classify_image2(image):
|
43 |
-
pipe2 = pipeline("image-classification", "SolubleFish/image_classification_convnext"
|
44 |
return pipe2(image)
|
45 |
def classify_image3(image):
|
46 |
-
pipe3 = pipeline("image-classification", "SolubleFish/image_classification_vit"
|
47 |
return pipe3(image)
|
48 |
|
49 |
|
|
|
37 |
# Image classification function
|
38 |
|
39 |
def classify_image1(image):
|
40 |
+
pipe1 = pipeline("image-classification", "SolubleFish/swin_transformer-finetuned-eurosat")
|
41 |
return pipe1(image)
|
42 |
def classify_image2(image):
|
43 |
+
pipe2 = pipeline("image-classification", "SolubleFish/image_classification_convnext")
|
44 |
return pipe2(image)
|
45 |
def classify_image3(image):
|
46 |
+
pipe3 = pipeline("image-classification", "SolubleFish/image_classification_vit")
|
47 |
return pipe3(image)
|
48 |
|
49 |
|