Spaces:
Runtime error
Runtime error
Trying ResNet changes
Browse files
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 =
|
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",
|