ktr008 commited on
Commit
1fec3ae
·
verified ·
1 Parent(s): 20a17f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from scipy.special import softmax
6
  # Load fine-tuned model and tokenizer
7
  MODEL_PATH = "ktr008/sentiment"
8
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
9
- tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
10
  config = AutoConfig.from_pretrained(MODEL_PATH)
11
 
12
  # Preprocess function
 
6
  # Load fine-tuned model and tokenizer
7
  MODEL_PATH = "ktr008/sentiment"
8
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
9
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH,timeout=600)
10
  config = AutoConfig.from_pretrained(MODEL_PATH)
11
 
12
  # Preprocess function