Spaces:

canadianjosieharrison
/

facade-material-classifier

Sleeping

canadianjosieharrison commited on Jan 9

Commit

d450a09

verified ·

1 Parent(s): f6ad690

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 pipe = pipeline(task="image-classification",
-                model="canadianjosieharrison/swinv2-large-patch4-window12-192-22k-baseline")
 gr.Interface.from_pipeline(pipe,
                            title="Facade material classification",

 from transformers import pipeline
 pipe = pipeline(task="image-classification",
+                model="canadianjosieharrison/swinv2-large-patch4-window12-192-22k-huggingface")
 gr.Interface.from_pipeline(pipe,
                            title="Facade material classification",