Hugging Face's logo

Spaces:
SrivarshiniGanesan
/
DeepStress
Runtime error

App Files Community
SrivarshiniGanesan commited on
Commit
204858a
·
verified ·
1 Parent(s): a50f398

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,9 +4,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
4
  import torch
5
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
6
 
7
- model_path = "./finetuned_model"
8
- model = AutoModelForSequenceClassification.from_pretrained(model_path)
9
- tokenizer = AutoTokenizer.from_pretrained(model_path)
10
 
11
  app = FastAPI(title="Stress Detection API", version="1.0")
12
 
 
4
  import torch
5
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
6
 
7
+ model = AutoModelForSequenceClassification.from_pretrained("SrivarshiniGanesan/finetuned-stress-model")
8
+ tokenizer = AutoTokenizer.from_pretrained("SrivarshiniGanesan/finetuned-stress-model")
 
9
 
10
  app = FastAPI(title="Stress Detection API", version="1.0")
11