Hugging Face's logo

Spaces:
Tokymin
/
Mood_Anxiety_Disorder_Classify
Sleeping

App Files Community
Tokymin commited on
Commit
c1571cc
·
1 Parent(s): 03212e7

更改requirements.txt文件的torch==1.12.0

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
4
 
5
  # 加载模型和tokenizer
6
- tokenizer = AutoTokenizer.from_pretrained("saved_models/model_20240302-214915_lr1e-05_optAdamW_lossBCEWithLogitsLoss_batch16_epoch10.pt")
7
- model = AutoModelForSequenceClassification.from_pretrained("saved_models/model_20240302-214915_lr1e-05_optAdamW_lossBCEWithLogitsLoss_batch16_epoch10.pt", num_labels=8)
8
  model.eval()
9
 
10
  def predict(text):
 
3
  import torch
4
 
5
  # 加载模型和tokenizer
6
+ tokenizer = AutoTokenizer.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify")
7
+ model = AutoModelForSequenceClassification.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify", num_labels=8)
8
  model.eval()
9
 
10
  def predict(text):