Matej commited on
Commit
eeaa8d0
1 Parent(s): fe6f0ef

load model from huggingface hub

Browse files
Files changed (1) hide show
  1. my_app.py +6 -2
my_app.py CHANGED
@@ -2,10 +2,14 @@ import gradio as gr
2
  import tensorflow as tf
3
  import numpy as np
4
  from PIL import Image
 
 
 
 
5
 
6
  # Load your trained models
7
- model1 = tf.keras.models.load_model('model/FoodVisionFineTuneAug/')
8
- model2 = tf.keras.models.load_model('model/FoodVisionFineTune/')
9
 
10
  with open('classes.txt', 'r') as f:
11
  classes = [line.strip() for line in f]
 
2
  import tensorflow as tf
3
  import numpy as np
4
  from PIL import Image
5
+ from huggingface_hub import from_pretrained_keras
6
+
7
+
8
+
9
 
10
  # Load your trained models
11
+ model1 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
12
+ model2 = from_pretrained_keras("ml-debi/EfficientNetB0-Food101")
13
 
14
  with open('classes.txt', 'r') as f:
15
  classes = [line.strip() for line in f]