MINHCT commited on
Commit
08386c1
1 Parent(s): 10177d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -6,6 +6,14 @@ from tensorflow.keras.preprocessing.sequence import pad_sequences # ensure that
6
  from tensorflow.keras.models import load_model # load a pre-trained Keras model saved in the Hierarchical Data Format (HDF5) file format
7
  import numpy as np # scientific computing in Python
8
  import streamlit as st
 
 
 
 
 
 
 
 
9
 
10
  # from . import SVM_Linear_Model
11
  import Logistic_Model
 
6
  from tensorflow.keras.models import load_model # load a pre-trained Keras model saved in the Hierarchical Data Format (HDF5) file format
7
  import numpy as np # scientific computing in Python
8
  import streamlit as st
9
+ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
10
+
11
+ model_name = "MINHCT/Classification"
12
+
13
+ # Tải tokenizer và mô hình từ Hugging Face Hub
14
+ # Tải tokenizer và mô hình từ Hugging Face Hub
15
+ tokenizer = AutoTokenizer.from_pretrained(vectorizer.joblib)
16
+ model = AutoModelForSequenceClassification.from_pretrained(Logistic_Model.joblib)
17
 
18
  # from . import SVM_Linear_Model
19
  import Logistic_Model