samithva commited on
Commit
da03bb7
·
verified ·
1 Parent(s): d8f01cc

Update app.py

Browse files

change model path

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,12 +5,12 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
 
6
  # Load your model and tokenizer
7
  model = AutoModelForSequenceClassification.from_pretrained(
8
- "./final_model",
9
  # load_in_8bit=True, # Use if you want to load in 8-bit quantized format
10
  # torch_dtype=torch.float16, # Use appropriate dtype based on your GPU
11
  # device_map="cuda:0" # Automatically map model to available devices
12
  )
13
- tokenizer = AutoTokenizer.from_pretrained("./final_model")
14
 
15
  # Ensure the model is in evaluation mode
16
  model.eval()
 
5
 
6
  # Load your model and tokenizer
7
  model = AutoModelForSequenceClassification.from_pretrained(
8
+ "./",
9
  # load_in_8bit=True, # Use if you want to load in 8-bit quantized format
10
  # torch_dtype=torch.float16, # Use appropriate dtype based on your GPU
11
  # device_map="cuda:0" # Automatically map model to available devices
12
  )
13
+ tokenizer = AutoTokenizer.from_pretrained("./")
14
 
15
  # Ensure the model is in evaluation mode
16
  model.eval()