breadlicker45 commited on
Commit
e60e9e6
·
verified ·
1 Parent(s): 5061607

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
9
  if HF_TOKEN is None:
10
  raise ValueError("Hugging Face token not found. Please set the HF_TOKEN secret in your Space settings.")
11
 
12
- MODEL_ID = "breadlicker45/genderclass-test3"
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  print(f"Using device: {device}")
15
 
@@ -55,7 +55,7 @@ def classify_gender(text: str) -> dict:
55
 
56
  DESCRIPTION = """
57
  ## Bilingual Gender Classifier
58
- This is a demo for the model `bilingual-base-gender-v4.5`.
59
  Enter a sentence and the model will predict whether the text has a male, female, or non-binary.
60
  **Disclaimer:** This model, like any AI, can have biases and may not always be accurate.
61
  """
 
9
  if HF_TOKEN is None:
10
  raise ValueError("Hugging Face token not found. Please set the HF_TOKEN secret in your Space settings.")
11
 
12
+ MODEL_ID = "breadlicker45/multilingual-bert-gender-classification"
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  print(f"Using device: {device}")
15
 
 
55
 
56
  DESCRIPTION = """
57
  ## Bilingual Gender Classifier
58
+ This is a demo for the model `breadlicker45/multilingual-bert-gender-classification`.
59
  Enter a sentence and the model will predict whether the text has a male, female, or non-binary.
60
  **Disclaimer:** This model, like any AI, can have biases and may not always be accurate.
61
  """