Muhammad-Hassaan commited on
Commit
583303c
·
verified ·
1 Parent(s): 747e5f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import numpy as np
3
  from tensorflow.keras.preprocessing import image
4
  import pickle
 
5
 
6
  with open('shoe_prediction.pkl', 'rb') as f:
7
  loaded_model = pickle.load(f)
 
2
  import numpy as np
3
  from tensorflow.keras.preprocessing import image
4
  import pickle
5
+ from tensorflow.keras.applications.resnet50 import preprocess_input
6
 
7
  with open('shoe_prediction.pkl', 'rb') as f:
8
  loaded_model = pickle.load(f)