Spaces:
Sleeping
Sleeping
Matej
commited on
Commit
·
eeaa8d0
1
Parent(s):
fe6f0ef
load model from huggingface hub
Browse files
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 =
|
8 |
-
model2 =
|
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]
|