ml-debi commited on
Commit
8d8e13c
1 Parent(s): f973db1

Update my_app.py

Browse files
Files changed (1) hide show
  1. my_app.py +2 -1
my_app.py CHANGED
@@ -4,8 +4,9 @@ from huggingface_hub import from_pretrained_keras
4
  from tensorflow.keras import mixed_precision
5
 
6
  # Load your trained models
7
- model1 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
8
  #model2 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
 
9
 
10
  with open('classes.txt', 'r') as f:
11
  classes = [line.strip() for line in f]
 
4
  from tensorflow.keras import mixed_precision
5
 
6
  # Load your trained models
7
+ #model1 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
8
  #model2 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
9
+ model1 = from_pretrained_keras("ml-debi/test_model")
10
 
11
  with open('classes.txt', 'r') as f:
12
  classes = [line.strip() for line in f]