duuuuuuuden commited on
Commit
5d806fd
·
1 Parent(s): 217866d

New model version to code

Browse files
Files changed (1) hide show
  1. main_app.py +1 -1
main_app.py CHANGED
@@ -13,7 +13,7 @@ class Url(BaseModel):
13
  link: str
14
 
15
 
16
- MODEL = tf.keras.models.load_model('./models/20_cat_classes_model.h5')
17
  GRADIO_PATH = '/'
18
 
19
  app = FastAPI()
 
13
  link: str
14
 
15
 
16
+ MODEL = tf.keras.models.load_model('./models/20_cat_classes_model_v2.h5')
17
  GRADIO_PATH = '/'
18
 
19
  app = FastAPI()