Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|