Hugging Face's logo

Spaces:
Jfink09
/
TCA
Sleeping

App Files Community
Jfink09 commited on
Commit
5df8bc7
·
verified ·
1 Parent(s): 6b29252

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoModelForSequenceClassification
3
 
4
  # Update this variable with your model name from Hugging Face Hub
5
- MODEL_NAME = "model.pt"
6
 
7
  # Load the model (no tokenizer needed)
8
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
 
2
  from transformers import AutoModelForSequenceClassification
3
 
4
  # Update this variable with your model name from Hugging Face Hub
5
+ MODEL_NAME = "model"
6
 
7
  # Load the model (no tokenizer needed)
8
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)