Spaces:

Srihari1611
/

Gender_Classification

Runtime error

Srihari1611 commited on Jun 7, 2023

Commit

3ae4ede

1 Parent(s): 6650da5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 import gradio as gr
+import fastai
+from fastai.vision.all import *
+learn=load_learner("model.pkl")
 categories=('male','female')
 def classify_image(img):
     pred,idx,probs=learn.predict(img)