Hugging Face's logo

Spaces:
hannawh
/
ai-academy-group7
Sleeping

App Files Community
Hanna Hjelmeland commited on
Commit
4a49928
•
1 Parent(s): 4578655

Update model paths

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,16 +14,16 @@ first_model = AutoModelForSequenceClassification.from_pretrained(first_model_pat
14
  second_model_path = "models/second_model"
15
  second_model = AutoModelForSequenceClassification.from_pretrained(second_model_path)
16
 
17
- f_30_40_model_path = "models/FEMALE_30_40model"
18
  f_30_40_model = AutoModelForSequenceClassification.from_pretrained(f_30_40_model_path)
19
 
20
- f_40_55_model_path = "models/FEMALE_40_55model"
21
  f_40_55_model = AutoModelForSequenceClassification.from_pretrained(f_40_55_model_path)
22
 
23
- m_30_40_model_path = "models/MALE_30_40model"
24
  m_30_40_model = AutoModelForSequenceClassification.from_pretrained(m_30_40_model_path)
25
 
26
- m_40_55_model_path = "models/MALE_40_55model"
27
  m_40_55_model = AutoModelForSequenceClassification.from_pretrained(m_40_55_model_path)
28
 
29
  def classify_text(test_text, selected_model='Model 3'):
 
14
  second_model_path = "models/second_model"
15
  second_model = AutoModelForSequenceClassification.from_pretrained(second_model_path)
16
 
17
+ f_30_40_model_path = "models/FEMALE_30_40model_new"
18
  f_30_40_model = AutoModelForSequenceClassification.from_pretrained(f_30_40_model_path)
19
 
20
+ f_40_55_model_path = "models/FEMALE_40_55model_new"
21
  f_40_55_model = AutoModelForSequenceClassification.from_pretrained(f_40_55_model_path)
22
 
23
+ m_30_40_model_path = "models/MALE_30_40model_new"
24
  m_30_40_model = AutoModelForSequenceClassification.from_pretrained(m_30_40_model_path)
25
 
26
+ m_40_55_model_path = "models/MALE_40_55model_new"
27
  m_40_55_model = AutoModelForSequenceClassification.from_pretrained(m_40_55_model_path)
28
 
29
  def classify_text(test_text, selected_model='Model 3'):