Spaces:

Srihari1611
/

Gender_Classification

Runtime error

Srihari1611 commited on Jun 7, 2023

Commit

56467b5

1 Parent(s): 9b1acb1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
-import fastai
-from fastai.vision.all import *
-learn=load_learner("gender_classification_model.h5")
 categories=('male','female')
 def classify_image(img):
     pred,idx,probs=learn.predict(img)

 import gradio as gr
+import tensorflow as tf
+from tensorflow import keras
+learn=tf.keras.models.load_model("gender_classification_model.h5")
 categories=('male','female')
 def classify_image(img):
     pred,idx,probs=learn.predict(img)