mxiean commited on
Commit
9820dbc
·
verified ·
1 Parent(s): b452ab5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def load_models():
15
  return {
16
  "style_classifier": pipeline(
17
  "image-classification",
18
- model="dima806/interior_design_style_classification"
19
  ),
20
  "advisor": pipeline("text2text-generation", model="google/flan-t5-small")
21
  }
 
15
  return {
16
  "style_classifier": pipeline(
17
  "image-classification",
18
+ model="playrobin/furniture-styles"
19
  ),
20
  "advisor": pipeline("text2text-generation", model="google/flan-t5-small")
21
  }