Ahsen Khaliq
commited on
Commit
•
e0f0523
1
Parent(s):
806d3da
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ import gradio as gr
|
|
17 |
|
18 |
mm = convnext.ConvNeXtBase()
|
19 |
|
20 |
-
def
|
21 |
img = img
|
22 |
imm = keras.applications.imagenet_utils.preprocess_input(img, mode='torch')
|
23 |
image_input = tf.expand_dims(tf.image.resize(imm, mm.input_shape[1:3]), 0)
|
|
|
17 |
|
18 |
mm = convnext.ConvNeXtBase()
|
19 |
|
20 |
+
def inference(img):
|
21 |
img = img
|
22 |
imm = keras.applications.imagenet_utils.preprocess_input(img, mode='torch')
|
23 |
image_input = tf.expand_dims(tf.image.resize(imm, mm.input_shape[1:3]), 0)
|