svnmurali1 commited on
Commit
e2826ac
·
verified ·
1 Parent(s): 922a0c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -15
app.py CHANGED
@@ -2,21 +2,6 @@ import gradio as gr
2
  import os
3
  from svnm.models import GenderDetection
4
  from datetime import datetime
5
- from tensorflow.keras.saving import register_keras_serializable
6
- from tensorflow.keras import layers
7
- import tensorflow as tf
8
- @register_keras_serializable()
9
- class ColorScaler(layers.Layer):
10
- def __init__(self,**kwargs):
11
- super(ColorScaler,self).__init__(**kwargs)
12
- def call(self, x):
13
- x = tf.image.rgb_to_grayscale(x) # Convert to grayscale
14
- x = tf.image.grayscale_to_rgb(x) # Convert back to RGB
15
- return x
16
- def get_config(self):
17
- config=super(ColorScaler,self).get_config()
18
- return config
19
-
20
  # Create a directory for saving uploaded images
21
  UPLOAD_DIR = "uploadimages"
22
  os.makedirs(UPLOAD_DIR, exist_ok=True)
 
2
  import os
3
  from svnm.models import GenderDetection
4
  from datetime import datetime
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  # Create a directory for saving uploaded images
6
  UPLOAD_DIR = "uploadimages"
7
  os.makedirs(UPLOAD_DIR, exist_ok=True)