CS473 commited on
Commit
cde5530
·
1 Parent(s): ed8c121

Trying ResNet changes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
3
  import numpy as np
4
  import tensorflow as tf
5
  import gradio as gr
6
-
7
  import cv2
8
  import keras
9
  from keras import Sequential
@@ -25,7 +25,7 @@ class PaceModel:
25
  self.create_architecture()
26
 
27
  def create_base_model(self):
28
- self.base_model = tf.keras.applications.ResNet50(
29
  include_top=False,
30
  input_shape=(self.height, self.width, self.channels),
31
  pooling="avg",
 
3
  import numpy as np
4
  import tensorflow as tf
5
  import gradio as gr
6
+ from keras.applications.resnet50 import ResNet50
7
  import cv2
8
  import keras
9
  from keras import Sequential
 
25
  self.create_architecture()
26
 
27
  def create_base_model(self):
28
+ self.base_model = ResNet50(
29
  include_top=False,
30
  input_shape=(self.height, self.width, self.channels),
31
  pooling="avg",