Jaesung Huh commited on
Commit
c2e5f66
β€’
1 Parent(s): 931ef66

remove model

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. gender_classifier.model +0 -3
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  import torch
3
  from model import ECAPA_gender
 
 
4
  # Load the model
5
  model = ECAPA_gender.from_pretrained("JaesungHuh/ecapa-gender")
6
- # model.load_state_dict(torch.load("gender_classifier.model", map_location="cpu"))
7
-
8
  model.eval()
9
 
10
  def predict_gender(filepath):
 
1
  import gradio as gr
2
  import torch
3
  from model import ECAPA_gender
4
+
5
+
6
  # Load the model
7
  model = ECAPA_gender.from_pretrained("JaesungHuh/ecapa-gender")
 
 
8
  model.eval()
9
 
10
  def predict_gender(filepath):
gender_classifier.model DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:e0e87253fe40aeef61b33e0fbe0859d9407e8bed92c8512cbaa28593b5f98e13
3
- size 62042234