OmerFarooq commited on
Commit
509b38e
1 Parent(s): ee191cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -7,6 +7,13 @@ import PIL
7
  from PIL import Image
8
  import gradio as gr
9
 
 
 
 
 
 
 
 
10
  #function for creating model
11
  #returns model, its inputs, Xception's last conv output, the whole model's outputs
12
  def create_model_mod():
 
7
  from PIL import Image
8
  import gradio as gr
9
 
10
+ base_model = keras.applications.Xception(
11
+ # weights = "../input/xception/xception_weights_tf_dim_ordering_tf_kernels_notop.h5",
12
+ input_shape = (160,160,3),
13
+ include_top = False,)
14
+
15
+ base_model.trainable = False
16
+
17
  #function for creating model
18
  #returns model, its inputs, Xception's last conv output, the whole model's outputs
19
  def create_model_mod():